Spaces:
Running
Running
fix suffix model repo
Browse files
app.py
CHANGED
@@ -61,7 +61,8 @@ def quantize_model(
|
|
61 |
try:
|
62 |
model_name = model_id.split("/")[-1]
|
63 |
username = whoami(oauth_token.token)["name"]
|
64 |
-
|
|
|
65 |
library_name = TasksManager.infer_library_from_model(model_id, token=oauth_token.token)
|
66 |
|
67 |
if library_name == "diffusers":
|
|
|
61 |
try:
|
62 |
model_name = model_id.split("/")[-1]
|
63 |
username = whoami(oauth_token.token)["name"]
|
64 |
+
suffix = f"{dtype}" if model_name.endswith("openvino") else f"openvino-{dtype}"
|
65 |
+
new_repo_id = f"{username}/{model_name}-{suffix}"
|
66 |
library_name = TasksManager.infer_library_from_model(model_id, token=oauth_token.token)
|
67 |
|
68 |
if library_name == "diffusers":
|