From d1efac58766631c0417c38b1416cc552345ffc40 Mon Sep 17 00:00:00 2001 From: junsongc Date: Fri, 22 Nov 2024 12:48:14 +0800 Subject: [PATCH] fix 'fcntl' bug in inference. Signed-off-by: junsongc --- app/sana_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/app/sana_pipeline.py b/app/sana_pipeline.py index af6aa9b..5a20f14 100644 --- a/app/sana_pipeline.py +++ b/app/sana_pipeline.py @@ -26,7 +26,7 @@ from diffusion import DPMS, FlowEuler -from diffusion.data.datasets.utils import * +from diffusion.data.datasets.utils import ASPECT_RATIO_512_TEST, ASPECT_RATIO_1024_TEST, ASPECT_RATIO_2048_TEST from diffusion.model.builder import build_model, get_tokenizer_and_text_encoder, get_vae, vae_decode from diffusion.model.utils import prepare_prompt_ar, resize_and_crop_tensor from diffusion.utils.config import SanaConfig