Fix style (#10478)

fix
2025-01-07 11:06:36 +05:30
parent 4f5e3e35d2
commit 661bde0ff2
28 changed files with 69 additions and 57 deletions
@@ -303,7 +303,9 @@ def save_blip_diffusion_model(model, args):
    qformer = get_qformer(model)
    qformer.eval()

-    text_encoder = ContextCLIPTextModel.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", subfolder="text_encoder")
+    text_encoder = ContextCLIPTextModel.from_pretrained(
+        "stable-diffusion-v1-5/stable-diffusion-v1-5", subfolder="text_encoder"
+    )
    vae = AutoencoderKL.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", subfolder="vae")
    unet = UNet2DConditionModel.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", subfolder="unet")
    vae.eval()