Update README.md
Browse files
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
|
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
|
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
|
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 |
|