Vae weight path none check (#784)

nod-ai · Jul 21, 2024 · aa0ac2b · aa0ac2b
1 parent 2e9de46
commit aa0ac2b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/models/turbine_models/custom_models/sd_inference/sd_pipeline.py b/models/turbine_models/custom_models/sd_inference/sd_pipeline.py
@@ -343,7 +343,8 @@ def __init__(
             self.scheduler_device = self.map["unet"]["device"]
             self.scheduler_driver = self.map["unet"]["driver"]
             self.scheduler_target = self.map["unet"]["target"]
-            self.map["vae"]["export_args"]["external_weight_path"] = vae_weight_path
+            if vae_weight_path is not None:
+                self.map["vae"]["export_args"]["external_weight_path"] = vae_weight_path
         elif not self.is_sd3:
             self.tokenizer = CLIPTokenizer.from_pretrained(
                 self.base_model_name, subfolder="tokenizer"