From 88bc9cc07dc0e217478f5d042cd4332cd7733033 Mon Sep 17 00:00:00 2001 From: lawrence-cj Date: Thu, 28 Nov 2024 03:21:20 +0800 Subject: [PATCH] lint; Signed-off-by: lawrence-cj --- scripts/inference_dpg.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/scripts/inference_dpg.py b/scripts/inference_dpg.py index 3e290b5..16364c8 100644 --- 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