Upload 9 files

app.py

@@ -55,6 +55,7 @@ In particular, conversion of FLUX.1 or SD 3.5 is almost impossible in CPU space.
             with gr.Row():
                 is_upload_sf = gr.Checkbox(label="Upload single safetensors file into new repo", value=False)
                 is_private = gr.Checkbox(label="Create private repo", value=True)
+                gated = gr.Radio(label="Create gated repo", info="Gated repo must be public", choices=["auto", "manual", "False"], value="False")
             with gr.Row():
                 is_overwrite = gr.Checkbox(label="Overwrite repo", value=HF_OW)
                 is_pr = gr.Checkbox(label="Create PR", value=HF_PR)
@@ -178,8 +179,7 @@ In particular, conversion of FLUX.1 or SD 3.5 is almost impossible in CPU space.
             sd35_run_button = gr.Button(value="Submit", variant="primary")
         adv_args = gr.Textbox(label="Advanced arguments", value="", visible=False)
         with gr.Group():
-            with gr.Row():
-                repo_urls = gr.CheckboxGroup(visible=False, choices=[], value=[])
+            repo_urls = gr.CheckboxGroup(visible=False, choices=[], value=[])
             output_md = gr.Markdown(label="Output", value="<br><br>", elem_classes="result")
             clear_button = gr.Button(value="Clear Output", variant="secondary")
         gr.DuplicateButton(value="Duplicate Space")
@@ -188,7 +188,7 @@ In particular, conversion of FLUX.1 or SD 3.5 is almost impossible in CPU space.
     gr.on(
         triggers=[sdxl_run_button.click],
         fn=convert_url_to_diffusers_repo,
-        inputs=[dl_url, hf_user, hf_repo, hf_token, civitai_key, is_private, is_overwrite, is_pr, is_upload_sf, repo_urls,
+        inputs=[dl_url, hf_user, hf_repo, hf_token, civitai_key, is_private, gated, is_overwrite, is_pr, is_upload_sf, repo_urls,
                 sdxl_dtype, sdxl_vae, sdxl_clip, flux_t5, sdxl_scheduler, sd15_ema, sd15_isize, sd15_sc, flux_base_repo, sdxl_mtype,
                 sdxl_lora1, sdxl_lora1s, sdxl_lora2, sdxl_lora2s, sdxl_lora3, sdxl_lora3s, sdxl_lora4, sdxl_lora4s, sdxl_lora5, sdxl_lora5s, adv_args],
         outputs=[repo_urls, output_md],
@@ -203,7 +203,7 @@ In particular, conversion of FLUX.1 or SD 3.5 is almost impossible in CPU space.
     gr.on(
         triggers=[sd15_run_button.click],
         fn=convert_url_to_diffusers_repo,
-        inputs=[dl_url, hf_user, hf_repo, hf_token, civitai_key, is_private, is_overwrite, is_pr, is_upload_sf, repo_urls,
+        inputs=[dl_url, hf_user, hf_repo, hf_token, civitai_key, is_private, gated, is_overwrite, is_pr, is_upload_sf, repo_urls,
                 sd15_dtype, sd15_vae, sd15_clip, flux_t5, sd15_scheduler, sd15_ema, sd15_isize, sd15_sc, flux_base_repo, sd15_mtype,
                 sd15_lora1, sd15_lora1s, sd15_lora2, sd15_lora2s, sd15_lora3, sd15_lora3s, sd15_lora4, sd15_lora4s, sd15_lora5, sd15_lora5s, adv_args],
         outputs=[repo_urls, output_md],
@@ -218,7 +218,7 @@ In particular, conversion of FLUX.1 or SD 3.5 is almost impossible in CPU space.
     gr.on(
         triggers=[flux_run_button.click],
         fn=convert_url_to_diffusers_repo,
-        inputs=[dl_url, hf_user, hf_repo, hf_token, civitai_key, is_private, is_overwrite, is_pr, is_upload_sf, repo_urls,
+        inputs=[dl_url, hf_user, hf_repo, hf_token, civitai_key, is_private, gated, is_overwrite, is_pr, is_upload_sf, repo_urls,
                 flux_dtype, flux_vae, flux_clip, flux_t5, flux_scheduler, sd15_ema, sd15_isize, sd15_sc, flux_base_repo, flux_mtype,
                 flux_lora1, flux_lora1s, flux_lora2, flux_lora2s, flux_lora3, flux_lora3s, flux_lora4, flux_lora4s, flux_lora5, flux_lora5s, adv_args],
         outputs=[repo_urls, output_md],
@@ -233,7 +233,7 @@ In particular, conversion of FLUX.1 or SD 3.5 is almost impossible in CPU space.
     gr.on(
         triggers=[sd35_run_button.click],
         fn=convert_url_to_diffusers_repo,
-        inputs=[dl_url, hf_user, hf_repo, hf_token, civitai_key, is_private, is_overwrite, is_pr, is_upload_sf, repo_urls,
+        inputs=[dl_url, hf_user, hf_repo, hf_token, civitai_key, is_private, gated, is_overwrite, is_pr, is_upload_sf, repo_urls,
                 sd35_dtype, sd35_vae, sd35_clip, sd35_t5, sd35_scheduler, sd15_ema, sd15_isize, sd15_sc, sd35_base_repo, sd35_mtype,
                 sd35_lora1, sd35_lora1s, sd35_lora2, sd35_lora2s, sd35_lora3, sd35_lora3s, sd35_lora4, sd35_lora4s, sd35_lora5, sd35_lora5s, adv_args],
         outputs=[repo_urls, output_md],

convert_url_to_diffusers_multi_gr.py

@@ -17,7 +17,7 @@ import shutil
 import gc
 import tempfile
 # also requires aria, gdown, peft, huggingface_hub, safetensors, transformers, accelerate, pytorch_lightning
-from utils import (get_token, set_token, is_repo_exists, is_repo_name, get_download_file, upload_repo)
+from utils import (get_token, set_token, is_repo_exists, is_repo_name, get_download_file, upload_repo, gate_repo)
 from sdutils import (SCHEDULER_CONFIG_MAP, get_scheduler_config, fuse_loras, DTYPE_DEFAULT, get_dtype, get_dtypes, get_model_type_from_key, get_process_dtype)
 
 
@@ -394,7 +394,8 @@ def convert_url_to_diffusers(url: str, civitai_key: str="", is_upload_sf: bool=F
         gc.collect()
 
 
-def convert_url_to_diffusers_repo(dl_url: str, hf_user: str, hf_repo: str, hf_token: str, civitai_key="", is_private: bool=True, is_overwrite: bool=False, is_pr: bool=False,
+def convert_url_to_diffusers_repo(dl_url: str, hf_user: str, hf_repo: str, hf_token: str, civitai_key="", is_private: bool=True,
+                                  gated: str="False", is_overwrite: bool=False, is_pr: bool=False,
                                   is_upload_sf: bool=False, urls: list=[], dtype: str="fp16", vae: str="", clip: str="", t5: str="", scheduler: str="Euler a",
                                   ema: bool=True, image_size: str="768", safety_checker: bool=False,
                                   base_repo: str="", mtype: str="", lora1: str="", lora1s=1.0, lora2: str="", lora2s=1.0, lora3: str="", lora3s=1.0,
@@ -404,6 +405,7 @@ def convert_url_to_diffusers_repo(dl_url: str, hf_user: str, hf_repo: str, hf_to
         if not civitai_key and os.environ.get("CIVITAI_API_KEY"): civitai_key = os.environ.get("CIVITAI_API_KEY") # default Civitai API key
         if not hf_token and os.environ.get("HF_TOKEN"): hf_token = os.environ.get("HF_TOKEN") # default HF write token
         if not hf_user: raise gr.Error(f"Invalid user name: {hf_user}")
+        if gated != "False" and is_private: raise gr.Error(f"Gated repo must be public")
         set_token(hf_token)
         lora_dict = {lora1: lora1s, lora2: lora2s, lora3: lora3s, lora4: lora4s, lora5: lora5s}
         new_path = convert_url_to_diffusers(dl_url, civitai_key, is_upload_sf, dtype, vae, clip, t5, scheduler, ema, image_size, safety_checker, base_repo, mtype, lora_dict, is_local)
@@ -413,6 +415,7 @@ def convert_url_to_diffusers_repo(dl_url: str, hf_user: str, hf_repo: str, hf_to
         if not is_repo_name(new_repo_id): raise gr.Error(f"Invalid repo name: {new_repo_id}")
         if not is_overwrite and is_repo_exists(new_repo_id) and not is_pr: raise gr.Error(f"Repo already exists: {new_repo_id}")
         repo_url = upload_repo(new_repo_id, new_path, is_private, is_pr)
+        gate_repo(new_repo_id, gated)
         safe_clean(new_path)
         if not urls: urls = []
         urls.append(repo_url)

utils.py

@@ -211,6 +211,18 @@ def upload_repo(repo_id: str, dir_path: str, is_private: bool, is_pr: bool=False
         return ""
 
 
+def gate_repo(repo_id: str, gated_str: str, repo_type: str="model"):
+    hf_token = get_token()
+    api = HfApi(token=hf_token)
+    try:
+        if gated_str == "auto": gated = "auto"
+        elif gated_str == "manual": gated = "manual"
+        else: gated = False
+        api.update_repo_settings(repo_id=repo_id, gated=gated, repo_type=repo_type, token=hf_token)
+    except Exception as e:
+        print(f"Error: Failed to update settings {repo_id}. {e}")
+
+
 HF_SUBFOLDER_NAME = ["None", "user_repo"]