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,
|