Spaces:

diffusers
/

compute-pipeline-size

Running

sayakpaul HF staff commited on Mar 7

Commit

1fad965

•

1 Parent(s): 7a40f6e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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", "tokenizer_2"]
 def format_size(num: int) -> str:

 import gradio as gr
 import json
+COMPONENT_FILTER = ["scheduler", "safety_checker", "feature_extractor", "tokenizer", "tokenizer_2"]
 def format_size(num: int) -> str: