SD quick fix export (#160)

fix export Co-authored-by: Jingya Huang <[email protected]>
huggingface · Aug 5, 2023 · c675fc0 · c675fc0
1 parent 424ec34
commit c675fc0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/neuron/modeling_diffusion.py b/optimum/neuron/modeling_diffusion.py
@@ -368,7 +368,7 @@ def _from_transformers(
             task = TasksManager.infer_task_from_model(cls.auto_model_class)
 
         # mandatory shapes
-        input_shapes = normalize_stable_diffusion_input_shapes(task, kwargs_shapes)
+        input_shapes = normalize_stable_diffusion_input_shapes(kwargs_shapes)
 
         # Get compilation arguments
         auto_cast_type = None if auto_cast is None else auto_cast_type