Spaces:

optimum
/

neuron-exporter

Running

badaoui HF Staff commited on Oct 11

Commit

cb59d1a

verified ·

1 Parent(s): e675f86

Update optimum_neuron_export.py

Files changed (1) hide show

optimum_neuron_export.py CHANGED Viewed

@@ -183,7 +183,6 @@ def export(model_id: str, task_or_pipeline: str, model_type: str, folder: str):
             token=HF_TOKEN,
             library_name=model_type,
             tensor_parallel_size=4,
-            cpu_backend=True,
             model=model,
             **inputs,
         )

             token=HF_TOKEN,
             library_name=model_type,
             tensor_parallel_size=4,
             model=model,
             **inputs,
         )