Set missing arg in SD tank generation

2026-04-03 03:00:17 -04:00 · 2023-03-21 14:12:42 -05:00
parent 7dbbb1726a
commit 57e99885e1
1 changed files with 1 additions and 0 deletions
--- a/tank/generate_sharktank.py
+++ b/tank/generate_sharktank.py
@@ -75,6 +75,7 @@ def save_torch_model(torch_model_list, local_tank_cache):
                            width=512,
                            height=512,
                            use_base_vae=False,
+                            custom_vae="",
                            debug=True,
                            sharktank_dir=local_tank_cache,
                            generate_vmfb=False,