Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix INC CLI #426

Merged
merged 13 commits into from
Sep 8, 2023
Merged

Fix INC CLI #426

Changes from all commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
17c5906
add vae image processor
echarlaix Sep 5, 2023
b00ec48
fix tests
echarlaix Sep 6, 2023
bbec65a
add test
echarlaix Sep 6, 2023
1a7e27d
add optimum min version
echarlaix Sep 7, 2023
f093353
add MPT support
echarlaix Sep 7, 2023
d8b451b
remove deprecated in optimum 1.13
echarlaix Sep 7, 2023
86ba0b3
Merge branch 'add-vae-processor' into mpt
echarlaix Sep 8, 2023
e1eeff8
set version optimum
echarlaix Sep 8, 2023
9f63bcf
fix
echarlaix Sep 8, 2023
f910ac0
patch model during export
echarlaix Sep 8, 2023
de5f4e9
update optimum latest compatible version
echarlaix Sep 8, 2023
5a67613
fix INC CLI
echarlaix Sep 8, 2023
9c4afd8
Merge branch 'main' into fix-inc-cli
echarlaix Sep 8, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions optimum/commands/neural_compressor/quantize.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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.",
)
Expand Down Expand Up @@ -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.",
Expand Down
Loading