danaevan commited on
Commit
3a8b815
·
verified ·
1 Parent(s): 8113fde

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -6
README.md CHANGED
@@ -7,13 +7,10 @@ language:
7
  tags:
8
  - Deci AI
9
  - DeciDiffusion
10
- datasets:
11
- - laion/laion-art
12
- - laion/laion2B-en
13
  ---
14
  # DeciDiffusion 2.0
15
 
16
- DeciDiffusion 2.0 is a 732 million parameter text-to-image latent diffusion model trained on the LAION-v2 dataset and fine-tuned on the LAION-ART dataset. Advanced training techniques were used to speed up training, improve training performance, and achieve better inference quality.
17
 
18
  ## Model Details
19
 
@@ -40,10 +37,10 @@ DeciDiffusion 2.0, a state-of-the-art diffusion-based text-to-image generation m
40
 
41
  The model was trained in 4 phases:
42
 
43
- - **Phase 1:** Trained from scratch 1.28 million steps at resolution 256x256 on a 270 million sample subset of LAION-v2.
44
  - **Phase 2:** Trained from 870k steps at resolution 512x512 on the same dataset to learn more fine-detailed information.
45
  - **Phase 3:** Trained 65k steps with EMA, another learning rate scheduler, and more "qualitative" data.
46
- - **Phase 4:** Fine-tuning on a 2M sample subset of LAION-ART.
47
 
48
  ### Training Techniques
49
 
 
7
  tags:
8
  - Deci AI
9
  - DeciDiffusion
 
 
 
10
  ---
11
  # DeciDiffusion 2.0
12
 
13
+ DeciDiffusion 2.0 is a 732 million parameter text-to-image latent diffusion model. Advanced training techniques were used to speed up training, improve training performance, and achieve better inference quality.
14
 
15
  ## Model Details
16
 
 
37
 
38
  The model was trained in 4 phases:
39
 
40
+ - **Phase 1:** Trained from scratch 1.28 million steps at resolution 256x256.
41
  - **Phase 2:** Trained from 870k steps at resolution 512x512 on the same dataset to learn more fine-detailed information.
42
  - **Phase 3:** Trained 65k steps with EMA, another learning rate scheduler, and more "qualitative" data.
43
+ - **Phase 4:** Fine-tuning on a 2M sample dataset.
44
 
45
  ### Training Techniques
46