Spaces:

optimum
/

neuron-exporter

Sleeping

badaoui HF Staff commited on Sep 10

Commit

46d4b4c

verified ·

1 Parent(s): 825604b

Update optimum_neuron_export.py

Files changed (1) hide show

optimum_neuron_export.py CHANGED Viewed

@@ -167,7 +167,7 @@ def export_and_git_add(model_id: str, task_or_pipeline: str, model_type: str, fo
         raise Exception(f"❌ Unsupported task/pipeline: {task_or_pipeline}. Supported: {supported}")
     inputs = get_default_inputs(task_or_pipeline)
-    compiler_args = prepare_compiler_flags(auto_cast, auto_cast_type, optlevel, instance_type)
     yield f"🔧 Using default inputs: {inputs}"
     try:

         raise Exception(f"❌ Unsupported task/pipeline: {task_or_pipeline}. Supported: {supported}")
     inputs = get_default_inputs(task_or_pipeline)
+    compiler_args = prepare_compiler_flags()
     yield f"🔧 Using default inputs: {inputs}"
     try: