sayakpaul HF staff commited on
Commit
828df18
1 Parent(s): 601c9fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -78,7 +78,7 @@ def get_component_wise_memory(pipeline_id, token=None, variant=None, revision=No
78
 
79
  gr.Interface(
80
  title="Compute component-wise memory of a 🧨 Diffusers pipeline.",
81
- description="Sizes will be reported in GB.",
82
  fn=get_component_wise_memory,
83
  inputs=[
84
  gr.components.Textbox(lines=1, label="pipeline_id", info="Example: runwayml/stable-diffusion-v1-5"),
@@ -102,7 +102,7 @@ gr.Interface(
102
  examples=[
103
  ["runwayml/stable-diffusion-v1-5", None, "fp32", None, ".safetensors"],
104
  ["stabilityai/stable-diffusion-xl-base-1.0", None, "fp16", None, ".safetensors"],
105
- [""],
106
  ],
107
  theme=gr.themes.Soft(),
108
  allow_flagging=False,
 
78
 
79
  gr.Interface(
80
  title="Compute component-wise memory of a 🧨 Diffusers pipeline.",
81
+ description="Sizes will be reported in GB. It can handle pipelines where the text encoder checkpoints might be shared (PixArt, for example).",
82
  fn=get_component_wise_memory,
83
  inputs=[
84
  gr.components.Textbox(lines=1, label="pipeline_id", info="Example: runwayml/stable-diffusion-v1-5"),
 
102
  examples=[
103
  ["runwayml/stable-diffusion-v1-5", None, "fp32", None, ".safetensors"],
104
  ["stabilityai/stable-diffusion-xl-base-1.0", None, "fp16", None, ".safetensors"],
105
+ ["PixArt-alpha/PixArt-XL-2-1024-MS", None, "fp32", None, ".safetensors"],
106
  ],
107
  theme=gr.themes.Soft(),
108
  allow_flagging=False,