Fix test instantiation of scheduling model.

nod-ai · May 28, 2024 · b65b577 · b65b577
1 parent 21dd3be
commit b65b577
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/models/turbine_models/tests/sd_test.py b/models/turbine_models/tests/sd_test.py
@@ -71,6 +71,7 @@
     default_arguments["height"],
     default_arguments["width"],
     default_arguments["num_inference_steps"],
+    default_arguments["precision"],
 )