clementchadebec commited on
Commit
d859638
1 Parent(s): 0975702

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
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 [ADD ARXIV]() *by Clément Chadebec, Onur Tasar, Eyal Benaroche, and Benjamin Aubin.*
 
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)