fix INC CLI

huggingface · Sep 8, 2023 · 5a67613 · 5a67613
1 parent de5f4e9
commit 5a67613
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/commands/neural_compressor/quantize.py b/optimum/commands/neural_compressor/quantize.py
@@ -28,7 +28,7 @@ def parse_args_inc_quantize(parser: "ArgumentParser"):
     required_group = parser.add_argument_group("Required arguments")
     required_group.add_argument(
         "--model",
-        type=Path,
+        type=str,
         required=True,
         help="Path to the repository where the model to quantize is located.",
     )
@@ -83,7 +83,7 @@ def run(self):
 
         if task == "auto":
             try:
-                task = TasksManager.infer_task_from_model(str(model_id))
+                task = TasksManager.infer_task_from_model(model_id)
             except Exception as e:
                 return (
                     f"### Error: {e}. Please pass explicitely the task as it could not be infered.",