Spaces:
Running
on
A100
Running
on
A100
MekkCyber
commited on
Commit
·
4d833a2
1
Parent(s):
7f64e83
update api
Browse files
app.py
CHANGED
|
@@ -86,7 +86,7 @@ def save_model(model, model_name, quantization_type, group_size=128, username=No
|
|
| 86 |
repo_name = f"{username}/{model_name.split('/')[-1]}-torchao-{quantization_type.lower()}"
|
| 87 |
|
| 88 |
# Push to Hub
|
| 89 |
-
api = HfApi()
|
| 90 |
api.create_repo(repo_name, exist_ok=True)
|
| 91 |
api.upload_folder(
|
| 92 |
folder_path=tmpdirname,
|
|
|
|
| 86 |
repo_name = f"{username}/{model_name.split('/')[-1]}-torchao-{quantization_type.lower()}"
|
| 87 |
|
| 88 |
# Push to Hub
|
| 89 |
+
api = HfApi(token=auth_token.token)
|
| 90 |
api.create_repo(repo_name, exist_ok=True)
|
| 91 |
api.upload_folder(
|
| 92 |
folder_path=tmpdirname,
|