Revert "Add model manager tab for webui (#1359)" (#1362)

This reverts commit 9d1d1617d8.

apps/stable_diffusion/web/index.py

@@ -90,7 +90,6 @@ if __name__ == "__main__":
         upscaler_sendto_inpaint,
         upscaler_sendto_outpaint,
         lora_train_web,
-        model_web,
     )
 
     # init global sd pipeline and config
@@ -120,8 +119,6 @@ if __name__ == "__main__":
                 outpaint_web.render()
             with gr.TabItem(label="Upscaler", id=4):
                 upscaler_web.render()
-            with gr.TabItem(label="Model Base", id=5):
-                model_web.render()
 
         with gr.Tabs(visible=False) as experimental_tabs:
             with gr.TabItem(label="LoRA Training", id=5):

apps/stable_diffusion/web/ui/__init__.py

@@ -49,4 +49,3 @@ from apps.stable_diffusion.web.ui.upscaler_ui import (
     upscaler_sendto_outpaint,
 )
 from apps.stable_diffusion.web.ui.lora_train_ui import lora_train_web
-from apps.stable_diffusion.web.ui.model_base import model_web

apps/stable_diffusion/web/ui/img2img_ui.py

@@ -366,7 +366,7 @@ with gr.Blocks(title="Image-to-Image") as img2img_web:
                         elem_id="hf_model_id",
                         placeholder="Select 'None' in the Models dropdown on the left and enter model ID here e.g: SG161222/Realistic_Vision_V1.3, https://a88802436301955b3a.gradio.live",
                         value="",
-                        label="HuggingFace Model ID or Civitai model download URL",
+                        label="HuggingFace Model ID or Civitai model download url",
                         lines=3,
                     )
                     custom_vae = gr.Dropdown(

apps/stable_diffusion/web/ui/inpaint_ui.py

@@ -320,7 +320,7 @@ with gr.Blocks(title="Inpainting") as inpaint_web:
                         elem_id="hf_model_id",
                         placeholder="Select 'None' in the Models dropdown on the left and enter model ID here e.g: ghunkins/stable-diffusion-liberty-inpainting, https://civitai.com/api/download/models/3433",
                         value="",
-                        label="HuggingFace Model ID or Civitai model download URL",
+                        label="HuggingFace Model ID or Civitai model download url",
                         lines=3,
                     )
                     custom_vae = gr.Dropdown(

apps/stable_diffusion/web/ui/model_base.py

@@ -1,85 +0,0 @@
-import os
-import gradio as gr
-from PIL import Image
-import json
-from shark.iree_utils._common import run_cmd
-
-
-def get_civit_list(num_of_models=100):
-    cmd = f'curl https://civitai.com/api/v1/models\?limit\={num_of_models}\&types\=Checkpoint -H "Content-Type: application/json" -X GET'
-    raw_json, error_msg = run_cmd(cmd)
-    models = list(json.loads(raw_json).items())[0][1]
-    safe_models = [
-        safe_model for safe_model in models if not safe_model["nsfw"]
-    ]
-    version_id = 0  # Currently just using the first version.
-    safe_models = [
-        safe_model
-        for safe_model in safe_models
-        if safe_model["modelVersions"][version_id]["files"][0]["metadata"][
-            "format"
-        ]
-        == "SafeTensor"
-    ]
-    first_version_models = []
-    for model_iter in safe_models:
-        # The modelVersion would only keep the version name.
-        if (
-            model_iter["modelVersions"][version_id]["images"][0]["nsfw"]
-            != "None"
-        ):
-            continue
-        model_iter["modelVersions"][version_id]["modelName"] = model_iter[
-            "name"
-        ]
-        model_iter["modelVersions"][version_id]["rating"] = model_iter[
-            "stats"
-        ]["rating"]
-        model_iter["modelVersions"][version_id]["favoriteCount"] = model_iter[
-            "stats"
-        ]["favoriteCount"]
-        model_iter["modelVersions"][version_id]["downloadCount"] = model_iter[
-            "stats"
-        ]["downloadCount"]
-        first_version_models.append(model_iter["modelVersions"][version_id])
-    return first_version_models
-
-
-def get_image_from_model(model_json):
-    model_id = model_json["modelId"]
-    path = None
-    for img_info in model_json["images"]:
-        if img_info["nsfw"] == "None":
-            image_url = model_json["images"][0]["url"]
-            save_img = f"curl {image_url} -o /tmp/{model_id}.jpeg"
-            run_cmd(save_img)
-            path = f"/tmp/{model_id}.jpeg"
-            break
-    return path
-
-
-model_list = get_civit_list()
-with gr.Blocks() as model_web:
-    for model in model_list:
-        with gr.Row():
-            img_path = get_image_from_model(model)
-            if img_path is None:
-                continue
-            model_img = Image.open(img_path)
-            gr.Image(
-                value=model_img,
-                show_label=False,
-                interactive=False,
-                elem_id="top_logo",
-            ).style(width=300, height=300)
-            with gr.Column():
-                gr.Textbox(
-                    label=f'{model["modelName"]}',
-                    value=f'Rating: {model["rating"]}{os.linesep}Favorite Count: {model["favoriteCount"]}{os.linesep}Download Count: {model["downloadCount"]}{os.linesep}File Format: {model["files"][0]["metadata"]["format"]}',
-                    lines=4,
-                )
-                gr.Textbox(
-                    label="Download URL:",
-                    value=f'{model["files"][0]["downloadUrl"]}',
-                    lines=1,
-                )

apps/stable_diffusion/web/ui/outpaint_ui.py

@@ -329,9 +329,9 @@ with gr.Blocks(title="Outpainting") as outpaint_web:
                     )
                     hf_model_id = gr.Textbox(
                         elem_id="hf_model_id",
-                        placeholder="Select 'None' in the Models dropdown on the left and enter model ID here e.g: ghunkins/stable-diffusion-liberty-inpainting, https://civitai.com/api/download/models/3433",
+                        placeholder="Select 'None' in the Models dropdown on the left and enter model ID here e.g: ghunkins/stable-diffusion-liberty-inpainting",
                         value="",
-                        label="HuggingFace Model ID or Civitai model download URL",
+                        label="HuggingFace Model ID",
                         lines=3,
                     )
                     custom_vae = gr.Dropdown(
@@ -371,9 +371,9 @@ with gr.Blocks(title="Outpainting") as outpaint_web:
                         )
                         lora_hf_id = gr.Textbox(
                             elem_id="lora_hf_id",
-                            placeholder="Select 'None' in the Standlone LoRA weights dropdown on the left if you want to use a standalone HuggingFace model ID for LoRA here e.g: sayakpaul/sd-model-finetuned-lora-t4",
+                            placeholder="Select 'None' in the Standlone LoRA weights dropdown on the left if you want to use a standalone HuggingFace model ID for LoRA here e.g: sayakpaul/sd-model-finetuned-lora-t4, https://civitai.com/api/download/models/3433",
                             value="",
-                            label="HuggingFace Model ID",
+                            label="HuggingFace Model ID or Civitai model download url",
                             lines=3,
                         )
                 with gr.Accordion(label="Advanced Options", open=False):

apps/stable_diffusion/web/ui/txt2img_ui.py

@@ -295,7 +295,7 @@ with gr.Blocks(title="Text-to-Image") as txt2img_web:
                                 elem_id="hf_model_id",
                                 placeholder="Select 'None' in the Models dropdown on the left and enter model ID here e.g: SG161222/Realistic_Vision_V1.3, https://civitai.com/api/download/models/15236",
                                 value="",
-                                label="HuggingFace Model ID or Civitai model download URL",
+                                label="HuggingFace Model ID or Civitai model download url",
                                 lines=3,
                             )
                             custom_vae = gr.Dropdown(

apps/stable_diffusion/web/ui/upscaler_ui.py

@@ -323,7 +323,7 @@ with gr.Blocks(title="Upscaler") as upscaler_web:
                         elem_id="hf_model_id",
                         placeholder="Select 'None' in the Models dropdown on the left and enter model ID here e.g: SG161222/Realistic_Vision_V1.3, https://civitai.com/api/download/models/15236",
                         value="",
-                        label="HuggingFace Model ID or Civitai model download URL",
+                        label="HuggingFace Model ID or Civitai model download url",
                         lines=3,
                     )
                     custom_vae = gr.Dropdown(