Spaces:

medmekk
/

TorchAO_Quantization

Running on A100

MekkCyber commited on Oct 18, 2024

Commit

4d833a2

1 Parent(s): 7f64e83

update api

Files changed (1) hide show

app.py CHANGED Viewed

@@ -86,7 +86,7 @@ def save_model(model, model_name, quantization_type, group_size=128, username=No
                 repo_name = f"{username}/{model_name.split('/')[-1]}-torchao-{quantization_type.lower()}"
         # Push to Hub
-        api = HfApi()
         api.create_repo(repo_name, exist_ok=True)
         api.upload_folder(
             folder_path=tmpdirname,

                 repo_name = f"{username}/{model_name.split('/')[-1]}-torchao-{quantization_type.lower()}"
         # Push to Hub
+        api = HfApi(token=auth_token.token)
         api.create_repo(repo_name, exist_ok=True)
         api.upload_folder(
             folder_path=tmpdirname,