diff --git a/scripts/inference_dpg.py b/scripts/inference_dpg.py
index 3e290b5..16364c8 100644
--- a/scripts/inference_dpg.py
+++ b/scripts/inference_dpg.py
@@ -34,7 +34,12 @@
 warnings.filterwarnings("ignore")  # ignore warning
 
 from diffusion import DPMS, FlowEuler, SASolverSampler
-from diffusion.data.datasets.utils import ASPECT_RATIO_512_TEST, ASPECT_RATIO_1024_TEST, ASPECT_RATIO_2048_TEST, get_chunks
+from diffusion.data.datasets.utils import (
+    ASPECT_RATIO_512_TEST,
+    ASPECT_RATIO_1024_TEST,
+    ASPECT_RATIO_2048_TEST,
+    get_chunks,
+)
 from diffusion.model.builder import build_model, get_tokenizer_and_text_encoder, get_vae, vae_decode
 from diffusion.model.utils import prepare_prompt_ar
 from diffusion.utils.config import SanaConfig