none check for vae external weight path

nod-ai · Jul 23, 2024 · a6b777f · a6b777f
1 parent 32fc090
commit a6b777f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/turbine_models/custom_models/sd_inference/vae.py b/models/turbine_models/custom_models/sd_inference/vae.py
@@ -163,7 +163,7 @@ def export_vae_model(
     if dtype == torch.float16:
         vae_model = vae_model.half()
     mapper = {}
-    if not os.path.exists(external_weight_path):
+    if (external_weight_path is not None) and (not os.path.exists(external_weight_path)):
         utils.save_external_weights(
             mapper,
             vae_model,