Spaces:
Running
Running
enable export of private and gated repo
Browse files- app.py +3 -1
- requirements.txt +7 -4
app.py
CHANGED
@@ -41,7 +41,9 @@ def export(model_id: str, private_repo: bool, oauth_token: gr.OAuthToken):
|
|
41 |
model_name = model_id.split("/")[-1]
|
42 |
username = whoami(oauth_token.token)["name"]
|
43 |
new_repo_id = f"{username}/{model_name}-openvino"
|
|
|
44 |
task = TasksManager.infer_task_from_model(model_id)
|
|
|
45 |
if task not in _HEAD_TO_AUTOMODELS:
|
46 |
raise ValueError(
|
47 |
f"The task '{task}' is not supported, only {_HEAD_TO_AUTOMODELS.keys()} tasks are supported"
|
@@ -67,7 +69,7 @@ def export(model_id: str, private_repo: bool, oauth_token: gr.OAuthToken):
|
|
67 |
os.makedirs(folder)
|
68 |
try:
|
69 |
api.snapshot_download(repo_id=model_id, local_dir=folder, allow_patterns=["*.json"])
|
70 |
-
ov_model = eval(auto_model_class).from_pretrained(model_id, export=True, cache_dir=folder)
|
71 |
ov_model.save_pretrained(folder)
|
72 |
new_repo_url = api.create_repo(repo_id=new_repo_id, exist_ok=True, private=private_repo)
|
73 |
new_repo_id = new_repo_url.repo_id
|
|
|
41 |
model_name = model_id.split("/")[-1]
|
42 |
username = whoami(oauth_token.token)["name"]
|
43 |
new_repo_id = f"{username}/{model_name}-openvino"
|
44 |
+
# task = TasksManager.infer_task_from_model(model_id, token=oauth_token.token)
|
45 |
task = TasksManager.infer_task_from_model(model_id)
|
46 |
+
|
47 |
if task not in _HEAD_TO_AUTOMODELS:
|
48 |
raise ValueError(
|
49 |
f"The task '{task}' is not supported, only {_HEAD_TO_AUTOMODELS.keys()} tasks are supported"
|
|
|
69 |
os.makedirs(folder)
|
70 |
try:
|
71 |
api.snapshot_download(repo_id=model_id, local_dir=folder, allow_patterns=["*.json"])
|
72 |
+
ov_model = eval(auto_model_class).from_pretrained(model_id, export=True, cache_dir=folder, token=oauth_token.token)
|
73 |
ov_model.save_pretrained(folder)
|
74 |
new_repo_url = api.create_repo(repo_id=new_repo_id, exist_ok=True, private=private_repo)
|
75 |
new_repo_id = new_repo_url.repo_id
|
requirements.txt
CHANGED
@@ -1,6 +1,9 @@
|
|
1 |
huggingface_hub==0.23.4
|
2 |
-
transformers==4.42.4
|
3 |
-
optimum[diffusers]==1.21.2
|
4 |
-
optimum-intel[openvino]==1.18.1
|
5 |
gradio[oauth]>=4.37.2
|
6 |
-
gradio_huggingfacehub_search==0.0.6
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
huggingface_hub==0.23.4
|
|
|
|
|
|
|
2 |
gradio[oauth]>=4.37.2
|
3 |
+
gradio_huggingfacehub_search==0.0.6
|
4 |
+
transformers==4.42.4
|
5 |
+
diffusers==0.29.1
|
6 |
+
optimum==1.21.2
|
7 |
+
optimum-intel==1.18.1
|
8 |
+
openvino
|
9 |
+
nncf
|