Spaces:
Runtime error
Runtime error
Commit
·
9b6512a
1
Parent(s):
4bc008c
updateapp.py
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ pipe = pipe.to(device)
|
|
| 16 |
|
| 17 |
# load pruned model
|
| 18 |
pruned_pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
|
| 19 |
-
pruned_pipe.
|
| 20 |
hf_hub_download("zhangyang-0123/EcoDiffPrunedModels", "model/sdxl/sdxl.pkl"),
|
| 21 |
map_location="cpu",
|
| 22 |
)
|
|
@@ -51,6 +51,7 @@ css = """
|
|
| 51 |
header = """
|
| 52 |
# 🌱 Text-to-Image Generation with EcoDiff Pruned SD-XL (20% Pruning Ratio)
|
| 53 |
# Under Construction!!!
|
|
|
|
| 54 |
<div style="text-align: center; display: flex; justify-content: left; gap: 5px;">
|
| 55 |
<a href="https://arxiv.org/abs/2412.02852"><img src="https://img.shields.io/badge/ariXv-Paper-A42C25.svg" alt="arXiv"></a>
|
| 56 |
<a href="https://huggingface.co/zhangyang-0123/EcoDiffPrunedModels"><img src="https://img.shields.io/badge/🤗-Model-ffbd45.svg" alt="HuggingFace"></a>
|
|
|
|
| 16 |
|
| 17 |
# load pruned model
|
| 18 |
pruned_pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
|
| 19 |
+
pruned_pipe.unet = torch.load(
|
| 20 |
hf_hub_download("zhangyang-0123/EcoDiffPrunedModels", "model/sdxl/sdxl.pkl"),
|
| 21 |
map_location="cpu",
|
| 22 |
)
|
|
|
|
| 51 |
header = """
|
| 52 |
# 🌱 Text-to-Image Generation with EcoDiff Pruned SD-XL (20% Pruning Ratio)
|
| 53 |
# Under Construction!!!
|
| 54 |
+
|
| 55 |
<div style="text-align: center; display: flex; justify-content: left; gap: 5px;">
|
| 56 |
<a href="https://arxiv.org/abs/2412.02852"><img src="https://img.shields.io/badge/ariXv-Paper-A42C25.svg" alt="arXiv"></a>
|
| 57 |
<a href="https://huggingface.co/zhangyang-0123/EcoDiffPrunedModels"><img src="https://img.shields.io/badge/🤗-Model-ffbd45.svg" alt="HuggingFace"></a>
|