fix package import bugs in dpg metric;

Signed-off-by: lawrence-cj <[email protected]>
NVlabs · Nov 27, 2024 · dce3e3f · dce3e3f
1 parent c84e2ee
commit dce3e3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/inference_dpg.py b/scripts/inference_dpg.py
@@ -34,7 +34,7 @@
 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
+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