From dce3e3f126b1d5e20beebdafeac88218b7bf4058 Mon Sep 17 00:00:00 2001 From: junsong Date: Wed, 27 Nov 2024 22:59:12 +0800 Subject: [PATCH] fix package import bugs in dpg metric; Signed-off-by: lawrence-cj --- scripts/inference_dpg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/inference_dpg.py b/scripts/inference_dpg.py index f9dc460..3e290b5 100644 --- 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