Upload 5 files

app.py

@@ -44,6 +44,7 @@ It saves you the trouble of typing them in.<br>
         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)
         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)
         model_type = gr.Radio(label="Model type", choices=["v1", "v2"], value="v1")
         sample_size = gr.Radio(label="Sample size (px)", choices=[512, 768], value=768)
@@ -63,12 +64,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_sd,
-        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,
                  model_type, sample_size, ema],
         outputs=[repo_urls, output_md],

convert_url_to_diffusers_sd.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_sd_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_sd(url, civitai_key="", half=True, vae=None, scheduler="Euler", lora_dict={},
+def convert_url_to_diffusers_sd(url, civitai_key="", is_upload_sf=False, half=True, vae=None, scheduler="Euler", lora_dict={},
                                   model_type="v1", sample_size=768, ema="ema", progress=gr.Progress(track_tqdm=True)):
     progress(0, desc="Start converting...")
     temp_dir = "."
@@ -264,6 +272,10 @@ def convert_url_to_diffusers_sd(url, civitai_key="", half=True, vae=None, schedu
     if Path(new_repo_name).exists():
         save_readme_md(new_repo_name, url)
 
+    if not is_repo_name(new_file) and is_upload_sf:
+        import shutil
+        shutil.move(str(Path(new_file).resolve()), str(Path(new_repo_name, Path(new_file).name).resolve()))
+
     progress(1, desc="Converted.")
     return new_repo_name
 
@@ -300,7 +312,7 @@ def create_diffusers_repo(new_repo_id, diffusers_folder, progress=gr.Progress(tr
     return url
 
 
-def convert_url_to_diffusers_repo_sd(dl_url, new_repo_id, hf_token, civitai_key="", repo_urls=[], half=True, vae=None,
+def convert_url_to_diffusers_repo_sd(dl_url, new_repo_id, hf_token, civitai_key="", is_upload_sf=False, repo_urls=[], half=True, vae=None,
                                   scheduler="Euler", 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,
                                   model_type="v1", sample_size=768, ema="ema", progress=gr.Progress(track_tqdm=True)):
@@ -316,7 +328,7 @@ def convert_url_to_diffusers_repo_sd(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_sd(dl_url, civitai_key, half, vae, scheduler, lora_dict,
+    new_path = convert_url_to_diffusers_sd(dl_url, civitai_key, is_upload_sf, half, vae, scheduler, lora_dict,
                                             model_type, sample_size, ema)
     if not new_path: return gr.update(value=repo_urls, choices=repo_urls), gr.update(value="")
     repo_url = create_diffusers_repo(new_repo_id, new_path)