Upload 5 files

app.py

@@ -53,8 +53,9 @@ It saves you the trouble of typing them in.<br>
     with gr.Column():
         dl_url = gr.Textbox(label="URL to download", placeholder="https://...", value="", max_lines=1)
         repo_id = gr.Textbox(label="Your New Repo ID", placeholder="author/model", value="", max_lines=1)
-        hf_token = gr.Textbox(label="Your HF write token", placeholder="", value="", max_lines=1)
+        hf_token = gr.Textbox(label="Your HF write token", placeholder="hf_...", value="", max_lines=1)
         civitai_key = gr.Textbox(label="Your Civitai API Key (Optional)", info="If you download model from Civitai...", placeholder="", value="", max_lines=1)
+        is_upload_sf = gr.Checkbox(label="Upload single safetensors file into new repo", value=False)
         is_half = gr.Checkbox(label="Half precision", value=True)
         vae = gr.Dropdown(label="VAE", choices=vaes, value="", allow_custom_value=True)
         scheduler = gr.Dropdown(label="Scheduler (Sampler)", choices=schedulers, value="Euler a")
@@ -71,12 +72,12 @@ It saves you the trouble of typing them in.<br>
         run_button = gr.Button(value="Submit")
         repo_urls = gr.CheckboxGroup(visible=False, choices=[], value=None)
         output_md = gr.Markdown(label="Output")
-        gr.DuplicateButton(value="Duplicate Space")
+    gr.DuplicateButton(value="Duplicate Space")
 
     gr.on(
         triggers=[run_button.click],
         fn=convert_url_to_diffusers_repo,
-        inputs=[dl_url, repo_id, hf_token, civitai_key, repo_urls, is_half, vae, scheduler,
+        inputs=[dl_url, repo_id, hf_token, civitai_key, is_upload_sf, repo_urls, is_half, vae, scheduler,
                  lora1, lora1s, lora2, lora2s, lora3, lora3s, lora4, lora4s, lora5, lora5s],
         outputs=[repo_urls, output_md],
     )

convert_url_to_diffusers_sdxl.py

@@ -168,6 +168,10 @@ def save_readme_md(dir, url):
     if orig_name and orig_url:
        md = f"""---
 license: other
+language:
+- en
+library_name: diffusers
+pipeline_tag: text-to-image
 tags:
 - text-to-image
 ---
@@ -176,6 +180,10 @@ Converted from [{orig_name}]({orig_url}).
     else:
         md = f"""---
 license: other
+language:
+- en
+library_name: diffusers
+pipeline_tag: text-to-image
 tags:
 - text-to-image
 ---

convert_url_to_diffusers_sdxl_gr.py

@@ -169,6 +169,10 @@ def save_readme_md(dir, url):
     if orig_name and orig_url:
        md = f"""---
 license: other
+language:
+- en
+library_name: diffusers
+pipeline_tag: text-to-image
 tags:
 - text-to-image
 ---
@@ -177,6 +181,10 @@ Converted from [{orig_name}]({orig_url}).
     else:
         md = f"""---
 license: other
+language:
+- en
+library_name: diffusers
+pipeline_tag: text-to-image
 tags:
 - text-to-image
 ---
@@ -205,7 +213,7 @@ def fuse_loras(pipe, lora_dict={}, temp_dir=".", civitai_key=""):
     pipe.unload_lora_weights()
 
 
-def convert_url_to_diffusers_sdxl(url, civitai_key="", half=True, vae=None, scheduler="Euler a", lora_dict={}, progress=gr.Progress(track_tqdm=True)):
+def convert_url_to_diffusers_sdxl(url, civitai_key="", is_upload_sf=False, half=True, vae=None, scheduler="Euler a", lora_dict={}, progress=gr.Progress(track_tqdm=True)):
     progress(0, desc="Start converting...")
     temp_dir = "."
     new_file = get_download_file(temp_dir, url, civitai_key)
@@ -225,6 +233,10 @@ def convert_url_to_diffusers_sdxl(url, civitai_key="", half=True, vae=None, sche
             pipe = StableDiffusionXLPipeline.from_single_file(new_file, use_safetensors=True, torch_dtype=torch.float16)
         else:
             pipe = StableDiffusionXLPipeline.from_single_file(new_file, use_safetensors=True)
+        if is_upload_sf:
+            import shutil
+            shutil.move(str(Path(new_file).resolve()), str(Path(new_repo_name, Path(new_file).name).resolve()))
+
 
     new_vae_file = ""
     if vae:
@@ -289,7 +301,7 @@ def create_diffusers_repo(new_repo_id, diffusers_folder, progress=gr.Progress(tr
     return url
 
 
-def convert_url_to_diffusers_repo(dl_url, new_repo_id, hf_token, civitai_key="", repo_urls=[], half=True, vae=None,
+def convert_url_to_diffusers_repo(dl_url, new_repo_id, hf_token, civitai_key="", is_upload_sf=False, repo_urls=[], half=True, vae=None,
                                   scheduler="Euler a", lora1=None, lora1s=1.0, lora2=None, lora2s=1.0, lora3=None, lora3s=1.0,
                                   lora4=None, lora4s=1.0, lora5=None, lora5s=1.0, progress=gr.Progress(track_tqdm=True)):
     if hf_token and not os.environ.get("HF_TOKEN"): os.environ['HF_TOKEN'] = hf_token
@@ -304,7 +316,7 @@ def convert_url_to_diffusers_repo(dl_url, new_repo_id, hf_token, civitai_key="",
         return gr.update(value=repo_urls, choices=repo_urls), gr.update(value="")
     lora_dict = {lora1: lora1s, lora2: lora2s, lora3: lora3s, lora4: lora4s, lora5: lora5s}
     if None in lora_dict.keys(): del lora_dict[None]
-    new_path = convert_url_to_diffusers_sdxl(dl_url, civitai_key, half, vae, scheduler, lora_dict)
+    new_path = convert_url_to_diffusers_sdxl(dl_url, civitai_key, is_upload_sf, half, vae, scheduler, lora_dict)
     if not new_path: return ""
     repo_url = create_diffusers_repo(new_repo_id, new_path)
     if not repo_urls: repo_urls = []

local/convert_url_to_diffusers_sdxl.py

@@ -168,6 +168,10 @@ def save_readme_md(dir, url):
     if orig_name and orig_url:
        md = f"""---
 license: other
+language:
+- en
+library_name: diffusers
+pipeline_tag: text-to-image
 tags:
 - text-to-image
 ---
@@ -176,6 +180,10 @@ Converted from [{orig_name}]({orig_url}).
     else:
         md = f"""---
 license: other
+language:
+- en
+library_name: diffusers
+pipeline_tag: text-to-image
 tags:
 - text-to-image
 ---