Spaces:

optimum
/

neuron-exporter

Running

badaoui HF Staff commited on Oct 11

Commit

3092c7d

verified ·

1 Parent(s): cb59d1a

Update optimum_neuron_export.py

Files changed (1) hide show

optimum_neuron_export.py CHANGED Viewed

@@ -184,7 +184,7 @@ def export(model_id: str, task_or_pipeline: str, model_type: str, folder: str):
             library_name=model_type,
             tensor_parallel_size=4,
             model=model,
-            **inputs,
         )
     else:

             library_name=model_type,
             tensor_parallel_size=4,
             model=model,
+            **inputs_shapes,
         )
     else: