fix runner function name in old sd test.

eagarvey-amd · eagarvey-amd · commit 26d954c11f25 · 2024-07-15T15:48:00.000-05:00
diff --git a/models/turbine_models/tests/sd_test.py b/models/turbine_models/tests/sd_test.py
@@ -208,7 +208,7 @@ def testExportVaeModelDecode(self):
             current_args["hf_model_name"],
             current_args["external_weight_path"],
         )
-        torch_output = vae_runner.run_torch_vae_decode(
+        torch_output = vae_runner.run_torch_vae(
             current_args["hf_model_name"],
             "decode",
             example_input,

Original file line number	Diff line number	Diff line change
`@@ -208,7 +208,7 @@ def testExportVaeModelDecode(self):`
`208`	`208`	`current_args["hf_model_name"],`
`209`	`209`	`current_args["external_weight_path"],`
`210`	`210`	`)`
`211`		`- torch_output = vae_runner.run_torch_vae_decode(`
	`211`	`+ torch_output = vae_runner.run_torch_vae(`
`212`	`212`	`current_args["hf_model_name"],`
`213`	`213`	`"decode",`
`214`	`214`	`example_input,`