Update sd3_vae.py

nod-ai · Aug 1, 2024 · 1a41394 · 1a41394
1 parent 0ac6b64
commit 1a41394
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/turbine_models/custom_models/sd3_inference/sd3_vae.py b/models/turbine_models/custom_models/sd3_inference/sd3_vae.py
@@ -98,7 +98,7 @@ def export_vae_model(
         vae_model = vae_model.half()
     mapper = {}
     utils.save_external_weights(
-        mapper, vae_model, external_weights, external_weight_path, vae_harness=True
+        mapper, vae_model, external_weights, external_weight_path
     )
     if weights_only:
         return external_weight_path