Update app.py
Browse files
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,
|