hide cards for networks of incompatible stable diffusion version in Lora extra networks interface

3 files changed, 11 insertions, 2 deletions

diff --git a/modules/sd_models.py b/modules/sd_models.py
index 729f03d7..4d9382dd 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -290,6 +290,9 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer
         state_dict = get_checkpoint_state_dict(checkpoint_info, timer)

 

     model.is_sdxl = hasattr(model, 'conditioner')

+    model.is_sd2 = not model.is_sdxl and hasattr(model.cond_stage_model, 'model')

+    model.is_sd1 = not model.is_sdxl and not model.is_sd2

+

     if model.is_sdxl:

         sd_models_xl.extend_sdxl(model)

 

diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py
index 6c73998f..49612298 100644
--- a/modules/ui_extra_networks.py
+++ b/modules/ui_extra_networks.py
@@ -62,7 +62,8 @@ def get_single_card(page: str = "", tabname: str = "", name: str = ""):
     page = next(iter([x for x in extra_pages if x.name == page]), None)

 

     try:

-        item = page.create_item(name)

+        item = page.create_item(name, enable_filter=False)

+        page.items[name] = item

     except Exception as e:

         errors.display(e, "creating item for extra network")

         item = page.items.get(name)

diff --git a/modules/ui_extra_networks_user_metadata.py b/modules/ui_extra_networks_user_metadata.py
index 01ff4e4b..63d4b503 100644
--- a/modules/ui_extra_networks_user_metadata.py
+++ b/modules/ui_extra_networks_user_metadata.py
@@ -42,6 +42,9 @@ class UserMetadataEditor:
 

         return user_metadata

 

+    def create_extra_default_items_in_left_column(self):

+        pass

+

     def create_default_editor_elems(self):

         with gr.Row():

             with gr.Column(scale=2):

@@ -49,6 +52,8 @@ class UserMetadataEditor:
                 self.edit_description = gr.Textbox(label="Description", lines=4)

                 self.html_filedata = gr.HTML()

 

+                self.create_extra_default_items_in_left_column()

+

             with gr.Column(scale=1, min_width=0):

                 self.html_preview = gr.HTML()

 

@@ -111,7 +116,7 @@ class UserMetadataEditor:
 

         table = '<table class="file-metadata">' + "".join(f"<tr><th>{name}</th><td>{value}</td></tr>" for name, value in params) + '</table>'

 

-        return html.escape(name), user_metadata.get('description', ''), table, self.get_card_html(name), user_metadata.get('notes', ''),

+        return html.escape(name), user_metadata.get('description', ''), table, self.get_card_html(name), user_metadata.get('notes', '')

 

     def write_user_metadata(self, name, metadata):

         item = self.page.items.get(name, {})