Upload convert_repo_to_safetensors_gr.py

convert_repo_to_safetensors_gr.py

@@ -22,9 +22,10 @@ def get_token():
     return token
 
 def is_repo_exists(repo_id: str, repo_type: str="model"):
-    api = HfApi(token=get_token())
+    hf_token = get_token()
+    api = HfApi(token=hf_token)
     try:
-        if api.repo_exists(repo_id=repo_id, repo_type=repo_type): return True
+        if api.repo_exists(repo_id=repo_id, repo_type=repo_type, token=hf_token): return True
         else: return False
     except Exception as e:
         print(f"Error: Failed to connect {repo_id} ({repo_type}). {e}")
@@ -353,8 +354,9 @@ def convert_diffusers_to_safetensors(model_path, checkpoint_path, dtype="fp16"):
 
 
 def download_repo(repo_id, dir_path):
+    hf_token = get_token()
     try:
-        snapshot_download(repo_id=repo_id, local_dir=dir_path, token=get_token())
+        snapshot_download(repo_id=repo_id, local_dir=dir_path, token=hf_token)
     except Exception as e:
         print(f"Error: Failed to download {repo_id}. {e}")
         gr.Warning(f"Error: Failed to download {repo_id}. {e}")
@@ -363,13 +365,14 @@ def download_repo(repo_id, dir_path):
 
 def upload_safetensors_to_repo(filename, repo_id, repo_type, is_private, progress=gr.Progress(track_tqdm=True)):
     output_filename = Path(filename).name
-    api = HfApi(token=get_token())
+    hf_token = get_token()
+    api = HfApi(token=hf_token)
     try:
-        if not is_repo_exists(repo_id, repo_type): api.create_repo(repo_id, repo_type=repo_type, private=is_private)
+        if not is_repo_exists(repo_id, repo_type): api.create_repo(repo_id, repo_type=repo_type, token=hf_token, private=is_private)
         progress(0, desc="Start uploading...")
-        api.upload_file(path_or_fileobj=filename, path_in_repo=output_filename, repo_type=repo_type, repo_id=repo_id)
+        api.upload_file(path_or_fileobj=filename, path_in_repo=output_filename, repo_type=repo_type, token=hf_token, repo_id=repo_id)
         progress(1, desc="Uploaded.")
-        url = hf_hub_url(repo_id=repo_id, repo_type=repo_type, filename=output_filename, token=get_token())
+        url = hf_hub_url(repo_id=repo_id, repo_type=repo_type, filename=output_filename, token=hf_token)
     except Exception as e:
         print(f"Error: Failed to upload to {repo_id}. {e}")
         gr.Warning(f"Error: Failed to upload to {repo_id}. {e}")