fix 'fcntl' bug in inference.

Signed-off-by: junsongc <[email protected]>
NVlabs · Nov 22, 2024 · d1efac5 · d1efac5
1 parent 82ebd27
commit d1efac5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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