Update README.md
Browse files
README.md
CHANGED
|
@@ -11,7 +11,8 @@ license: cc-by-nc-nd-4.0
|
|
| 11 |
# ⚡ Flash Diffusion: FlashSDXL ⚡
|
| 12 |
|
| 13 |
|
| 14 |
-
Flash Diffusion is a diffusion distillation method proposed in [
|
|
|
|
| 15 |
This model is a **108M** LoRA distilled version of [SDXL](https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0) model that is able to generate images in **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 4 H100 GPUs (representing approximately a total of 176 GPU hours of training). Please refer to the [paper]() for further parameters details.
|
| 58 |
|
| 59 |
**Metrics on COCO 2014 validation (Table 3)**
|
| 60 |
- FID-10k: 21.62 (4 NFE)
|
|
|
|
| 11 |
# ⚡ Flash Diffusion: FlashSDXL ⚡
|
| 12 |
|
| 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 **108M** LoRA distilled version of [SDXL](https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0) model that is able to generate images in **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 4 H100 GPUs (representing approximately a total of 176 GPU hours of training). Please refer to the [paper](http://arxiv.org/abs/2406.02347) for further parameters details.
|
| 59 |
|
| 60 |
**Metrics on COCO 2014 validation (Table 3)**
|
| 61 |
- FID-10k: 21.62 (4 NFE)
|