fix tests

huggingface · Oct 22, 2024 · 3679d53 · 3679d53
1 parent 38cda01
commit 3679d53
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/exporters/test_export.py b/tests/exporters/test_export.py
@@ -152,7 +152,7 @@ def _neuronx_export(
         with NamedTemporaryFile("w") as output:
             try:
                 _, neuron_outputs = export(
-                    model=model,
+                    model_or_path=model,
                     config=neuron_config,
                     output=Path(output.name),
                     inline_weights_to_neff=inline_weights_to_neff,