clementchadebec
commited on
Commit
•
eb707ae
1
Parent(s):
7427aae
Update README.md
Browse files
README.md
CHANGED
@@ -11,7 +11,8 @@ inference: False
|
|
11 |
---
|
12 |
# ⚡ Flash Diffusion: FlashSD ⚡
|
13 |
|
14 |
-
Flash Diffusion is a diffusion distillation method proposed in [
|
|
|
15 |
This model is a **26.4M** LoRA distilled version of SD1.5 model that is able to generate images in **2-4 steps**. The main purpose of this model is to reproduce the main results of the paper.
|
16 |
See our [live demo](https://huggingface.co/spaces/jasperai/FlashPixart).
|
17 |
|
@@ -54,7 +55,7 @@ image = pipe(prompt, num_inference_steps=4, guidance_scale=0).images[0]
|
|
54 |
</p>
|
55 |
|
56 |
# Training Details
|
57 |
-
The model was trained for 20k iterations on 2 H100 GPUs (representing approx. a total **26 GPU hours** of training). Please refer to the [paper]() for further parameters details.
|
58 |
|
59 |
**Metrics on COCO 2017 validation set (Table 1)**
|
60 |
- FID-5k: 22.6 (2 NFE) / 22.5 (4 NFE)
|
|
|
11 |
---
|
12 |
# ⚡ Flash Diffusion: FlashSD ⚡
|
13 |
|
14 |
+
Flash Diffusion is a diffusion distillation method proposed in [Flash Diffusion: Accelerating Any Conditional
|
15 |
+
Diffusion Model for Few Steps Image Generation](http://arxiv.org/abs/2406.02347) *by Clément Chadebec, Onur Tasar, Eyal Benaroche, and Benjamin Aubin.*
|
16 |
This model is a **26.4M** LoRA distilled version of SD1.5 model that is able to generate images in **2-4 steps**. The main purpose of this model is to reproduce the main results of the paper.
|
17 |
See our [live demo](https://huggingface.co/spaces/jasperai/FlashPixart).
|
18 |
|
|
|
55 |
</p>
|
56 |
|
57 |
# Training Details
|
58 |
+
The model was trained for 20k iterations on 2 H100 GPUs (representing approx. a total **26 GPU hours** of training). Please refer to the [paper](http://arxiv.org/abs/2406.02347) for further parameters details.
|
59 |
|
60 |
**Metrics on COCO 2017 validation set (Table 1)**
|
61 |
- FID-5k: 22.6 (2 NFE) / 22.5 (4 NFE)
|