Update README.md
Browse files
README.md
CHANGED
@@ -45,7 +45,7 @@ The model was trained in 4 phases:
|
|
45 |
|
46 |
### Training Techniques
|
47 |
|
48 |
-
DeciDiffusion 1.0 was trained to be sample efficient, i.e. to
|
49 |
The following training techniques were used to that end:
|
50 |
|
51 |
- **[V-prediction](https://arxiv.org/pdf/2202.00512.pdf)**
|
@@ -69,7 +69,7 @@ The following techniques were used to shorten training time:
|
|
69 |
- **Batch:** 8192
|
70 |
- **Learning rate:** 1e-4
|
71 |
|
72 |
-
####
|
73 |
- **Hardware:** 8 x 8 x H100 (80gb)
|
74 |
- **Optimizer:** LAMB
|
75 |
- **Batch:** 6144
|
@@ -85,7 +85,7 @@ Given this skepticism about FID’s reliability, we chose to assess DeciDiffusio
|
|
85 |
For our study we chose 10 random prompts and for each prompt generated 3 images
|
86 |
by Stable Diffusion 1.5 configured to run for 50 iterations and 3 images by DeciDiffusion configured to run for 30 iterations.
|
87 |
|
88 |
-
We then presented 30 side by side comparisons to
|
89 |
|
90 |
According to the results, DeciDiffusion at 30 iterations exhibits an edge in aesthetics, but when it comes to prompt alignment, it’s on par with Stable Diffusion at 50 iterations.
|
91 |
|
|
|
45 |
|
46 |
### Training Techniques
|
47 |
|
48 |
+
DeciDiffusion 1.0 was trained to be sample efficient, i.e. to produce high-quality results using fewer diffusion timesteps during inference.
|
49 |
The following training techniques were used to that end:
|
50 |
|
51 |
- **[V-prediction](https://arxiv.org/pdf/2202.00512.pdf)**
|
|
|
69 |
- **Batch:** 8192
|
70 |
- **Learning rate:** 1e-4
|
71 |
|
72 |
+
#### Phases 2-4
|
73 |
- **Hardware:** 8 x 8 x H100 (80gb)
|
74 |
- **Optimizer:** LAMB
|
75 |
- **Batch:** 6144
|
|
|
85 |
For our study we chose 10 random prompts and for each prompt generated 3 images
|
86 |
by Stable Diffusion 1.5 configured to run for 50 iterations and 3 images by DeciDiffusion configured to run for 30 iterations.
|
87 |
|
88 |
+
We then presented 30 side by side comparisons to a group of professionals, who voted based on adherence to the prompt and aesthetic value.
|
89 |
|
90 |
According to the results, DeciDiffusion at 30 iterations exhibits an edge in aesthetics, but when it comes to prompt alignment, it’s on par with Stable Diffusion at 50 iterations.
|
91 |
|