Skip to content

Commit 97255f9

Browse files
committed
docs: update distilled_sd.md for the Vega model
1 parent 75c0b7f commit 97255f9

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

docs/distilled_sd.md

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
1-
# Running distilled models: SSD1B and SDx.x with tiny U-Nets
1+
# Running distilled models: SSD1B, Vega and SDx.x with tiny U-Nets
22

33
## Preface
44

5-
These models feature a reduced U-Net architecture. Unlike standard SDXL models, the SSD-1B U-Net contains only one middle block and fewer attention layers in its up- and down-blocks, resulting in significantly smaller file sizes. Using these models can reduce inference time by more than 33%. For more details, refer to Segmind's paper: https://arxiv.org/abs/2401.02677v1.
5+
These models feature a reduced U-Net architecture. Unlike standard SDXL models, the SSD-1B and Vega U-Net contains only one middle block and fewer attention layers in its up- and down-blocks, resulting in significantly smaller file sizes. Using these models can reduce inference time by more than 33%. For more details, refer to Segmind's paper: https://arxiv.org/abs/2401.02677v1.
66
Similarly, SD1.x- and SD2.x-style models with a tiny U-Net consist of only 6 U-Net blocks, leading to very small files and time savings of up to 50%. For more information, see the paper: https://arxiv.org/pdf/2305.15798.pdf.
77

88
## SSD1B
@@ -17,7 +17,17 @@ Useful LoRAs are also available:
1717
* https://huggingface.co/seungminh/lora-swarovski-SSD-1B/resolve/main/pytorch_lora_weights.safetensors
1818
* https://huggingface.co/kylielee505/mylcmlorassd/resolve/main/pytorch_lora_weights.safetensors
1919

20-
These files can be used out-of-the-box, unlike the models described in the next section.
20+
## Vega
21+
22+
Segmind's Vega model is available online here:
23+
24+
* https://huggingface.co/segmind/Segmind-Vega/resolve/main/segmind-vega.safetensors
25+
26+
VegaRT is an example for an LCM-LoRA:
27+
28+
* https://huggingface.co/segmind/Segmind-VegaRT/resolve/main/pytorch_lora_weights.safetensors
29+
30+
Both files can be used out-of-the-box, unlike the models described in next sections.
2131

2232

2333
## SD1.x, SD2.x with tiny U-Nets

0 commit comments

Comments
 (0)