envs

app.py

@@ -197,7 +197,7 @@ class ImageConductor:
         text_encoder = CLIPTextModel.from_pretrained("runwayml/stable-diffusion-v1-5", subfolder="text_encoder")
         vae          = AutoencoderKL.from_pretrained("runwayml/stable-diffusion-v1-5", subfolder="vae")
         inference_config = OmegaConf.load("configs/inference/inference.yaml")
-        unet = UNet3DConditionFlowModel.from_pretrained_2d("models/sd1-5/", subfolder="unet", unet_additional_kwargs=OmegaConf.to_container(inference_config.unet_additional_kwargs))
+        unet = UNet3DConditionFlowModel.from_pretrained_2d("models/sd1-5/", unet_additional_kwargs=OmegaConf.to_container(inference_config.unet_additional_kwargs))
 
         self.vae = vae