Fix CLI for vae_runner.py

nod-ai · Feb 27, 2024 · 0d5e913 · 0d5e913
1 parent d2c7ebb
commit 0d5e913
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/turbine_models/custom_models/sdxl_inference/vae_runner.py b/models/turbine_models/custom_models/sdxl_inference/vae_runner.py
@@ -148,7 +148,7 @@ def encode_inp(self, inp):
         print("generating torch output: ")
         from turbine_models.custom_models.sd_inference import utils
 
-        torch_output = run_torch_vae(args.hf_model_name, args.variant, example_input)
+        torch_output = run_torch_vae(args.hf_model_name, "", args.variant, example_input)
         print("TORCH OUTPUT:", torch_output, torch_output.shape, torch_output.dtype)
         err = utils.largest_error(torch_output, turbine_results)
         print("Largest Error: ", err)