lint;

Signed-off-by: lawrence-cj <[email protected]>
NVlabs · Nov 27, 2024 · 88bc9cc · 88bc9cc
1 parent 2e1f964
commit 88bc9cc
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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