Update app.py

app.py

@@ -2,7 +2,7 @@ from huggingface_hub import model_info, hf_hub_download
 import gradio as gr
 import json
 
-component_filter = ["scheduler", "safety_checker", "tokenizer"]
+COMPONENT_FILTER = ["scheduler", "safety_checker", "tokenizer"]
 
 
 def format_size(num: int) -> str:
@@ -48,8 +48,8 @@ def get_component_wise_memory(pipeline_id, token=None, variant=None, revision=No
     index_dict = load_model_index(pipeline_id, token=token, revision=revision)
 
     # Check if all the concerned components have the checkpoints in the requested "variant" and "extension".
-    index_filter = component_filter.copy()
-    index_filter.extend(["_class_name", "_diffusers_version"])
+    index_filter = COMPONENT_FILTER.copy()
+    index_filter.extend(["_class_name", "_diffusers_version", "force_zeros_for_empty_prompt", "add_watermarker"])
     for current_component in index_dict:
         if current_component not in index_filter:
             current_component_fileobjs = list(filter(lambda x: current_component in x.rfilename, files_in_repo))
@@ -86,10 +86,10 @@ def get_component_wise_memory(pipeline_id, token=None, variant=None, revision=No
 
     # Handle pipeline components.
     if is_text_encoder_shared:
-        component_filter.append("text_encoder")
+        COMPONENT_FILTER.append("text_encoder")
 
     for current_file in files_in_repo:
-        if all(substring not in current_file.rfilename for substring in component_filter):
+        if all(substring not in current_file.rfilename for substring in COMPONENT_FILTER):
             is_folder = len(current_file.rfilename.split("/")) == 2
             if is_folder and current_file.rfilename.split("/")[0] in index_dict:
                 selected_file = None