Upload 3 files

README.md

@@ -1,5 +1,5 @@
 ---
-title: Convert diffusers SDXL repo to single Safetensors
+title: Convert diffusers SDXL repo to single Safetensors V2
 emoji: 🐶
 colorFrom: yellow
 colorTo: red

app.py

@@ -5,7 +5,7 @@ os.environ['HF_OUTPUT_REPO'] = 'John6666/safetensors_converting_test'
 
 css = """"""
 
-with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", css=css) as demo:
+with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css=css, delete_cache=(60, 3600)) as demo:
     gr.Markdown(
             f"""
     - [A CLI version of this tool is available here](https://huggingface.co/spaces/John6666/convert_repo_to_safetensors/tree/main/local).

convert_repo_to_safetensors_gr.py

@@ -21,13 +21,13 @@ def get_token():
         token = ""
     return token
 
-def is_repo_exists(repo_id: str):
+def is_repo_exists(repo_id: str, repo_type: str="model"):
     api = HfApi(token=get_token())
     try:
-        if api.repo_exists(repo_id=repo_id): return True
+        if api.repo_exists(repo_id=repo_id, repo_type=repo_type): return True
         else: return False
     except Exception as e:
-        print(f"Error: Failed to connect {repo_id}. {e}")
+        print(f"Error: Failed to connect {repo_id} ({repo_type}). {e}")
         return True # for safe
 
 # =================#
@@ -356,6 +356,7 @@ def convert_diffusers_to_safetensors(model_path, checkpoint_path, dtype="fp16",
 
 def download_repo(repo_id, dir_path, progress=gr.Progress(track_tqdm=True)):
     try:
+        progress(0, desc="Start downloading...")
         snapshot_download(repo_id=repo_id, local_dir=dir_path, token=get_token())
     except Exception as e:
         print(f"Error: Failed to download {repo_id}. {e}")
@@ -366,11 +367,11 @@ def upload_safetensors_to_repo(filename, repo_id, repo_type, is_private, progres
     output_filename = Path(filename).name
     api = HfApi(token=get_token())
     try:
-        if not is_repo_exists(repo_id): 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, 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)
         progress(1, desc="Uploaded.")
-        url  = hf_hub_url(repo_id=repo_id, filename=output_filename)
+        url = hf_hub_url(repo_id=repo_id, repo_type=repo_type, filename=output_filename, token=get_token())
     except Exception as e:
         print(f"Error: Failed to upload to {repo_id}. {e}")
         return None
@@ -391,11 +392,11 @@ def convert_repo_to_safetensors_multi(repo_id, hf_token, files, urls, dtype="fp1
     if hf_token: HfFolder.save_token(hf_token)
     else: HfFolder.save_token(os.environ.get("HF_TOKEN"))
     if not newrepo_id: newrepo_id = os.environ.get("HF_OUTPUT_REPO")
-    file = convert_repo_to_safetensors(repo_id, dtype)
+    file = convert_repo_to_safetensors(repo_id, dtype, progress)
     if not urls: urls = []
     url = ""
     if is_upload:
-        url = upload_safetensors_to_repo(file, newrepo_id, repo_type, is_private)
+        url = upload_safetensors_to_repo(file, newrepo_id, repo_type, is_private, progress)
         if url: urls.append(url)
     md = ""
     for u in urls: