feat: added text2img CFG images to readme

AshishKumar4 · Jul 16, 2024 · 5644432 · 5644432
1 parent 11e4b31
commit 5644432
Show file tree

Hide file tree

Showing 7 changed files with 335 additions and 52 deletions.
diff --git a/Diffusion flax linen.ipynb b/Diffusion flax linen.ipynb
diff --git a/README.md b/README.md
@@ -201,7 +201,33 @@ plotImages(samples, dpi=300)
 
 ## Gallery
 
-### Images generated by DDPM Sampler in 1000 steps
+### Images generated by Euler Ancestral Sampler in 200 Steps [text2image with CFG]
+Images generated by the following prompts using classifier free guidance with guidance factor = 2:
+  - 'water tulip',
+  - 'a water lily',
+  - 'a water lily', 
+  - a water lily', 
+  - 'a photo of a marigold', 
+  - 'a water lily',
+  - 'a water lily',
+  - 'a photo of a lotus',
+  - 'a photo of a lotus',
+  - 'a photo of a lotus',
+  - 'a photo of a rose',
+  - 'a photo of a rose',
+  - 'a photo of a rose',
+  - 'a photo of a rose',
+  - 'a photo of a rose'
+
+![EulerA with CFG](images/text2img%20euler%20ancestral%201.png)
+
+### Images generated by Euler Ancestral Sampler in 200 Steps [text2image with CFG]
+Images generated by the following prompts using classifier free guidance with guidance factor = 4: 
+`'water tulip, a water lily, a water lily, a photo of a rose, a photo of a rose, a water lily, a water lily, a photo of a marigold, a photo of a marigold, a photo of a marigold, a water lily, a photo of a sunflower, a photo of a lotus, columbine, columbine, an orchid, an orchid, an orchid, a water lily, a water lily, a water lily, columbine, columbine, a photo of a sunflower, a photo of a sunflower, a photo of a sunflower, a photo of a lotus, a photo of a lotus, a photo of a marigold, a photo of a marigold, a photo of a rose, a photo of a rose, a photo of a rose, orange dahlia, orange dahlia, a lenten rose, a lenten rose, a water lily, a water lily, a water lily, a water lily, an orchid, an orchid, an orchid, hard-leaved pocket orchid, bird of paradise, bird of paradise, a photo of a lovely rose, a photo of a lovely rose, a photo of a globe-flower, a photo of a globe-flower, a photo of a lovely rose, a photo of a lovely rose, a photo of a ruby-lipped cattleya, a photo of a ruby-lipped cattleya, a photo of a lovely rose, a water lily, a osteospermum, a osteospermum, a water lily, a water lily, a water lily, a red rose, a red rose'`
+
+![EulerA with CFG](images/text2img%20euler%20ancestral%202.png)
+
+### Images generated by DDPM Sampler in 1000 steps [Unconditional]
 
 **Params**: 
 `Dataset: oxford_flowers102`
@@ -221,7 +247,7 @@ plotImages(samples, dpi=300)
 
 ![DDPM Sampler results](images/ddpm2.png)
 
-### Images generated by DDPM Sampler in 1000 steps
+### Images generated by DDPM Sampler in 1000 steps [Unconditional]
 
 **Params**: 
 `Dataset: oxford_flowers102`
@@ -241,7 +267,7 @@ plotImages(samples, dpi=300)
 
 ![DDPM Sampler results](images/ddpm.png)
 
-### Images generated by Heun Sampler in 10 steps (20 model inferences as Heun takes 2x inference steps)
+### Images generated by Heun Sampler in 10 steps (20 model inferences as Heun takes 2x inference steps) [Unconditional]
 
 **Params**: 
 `Dataset: oxford_flowers102`

diff --git a/images/text2img euler ancestral 1.png b/images/text2img euler ancestral 1.png
diff --git a/images/text2img euler ancestral 2.png b/images/text2img euler ancestral 2.png
diff --git a/example notebooks/edm tutorial.ipynb → tutorial notebooks/edm tutorial.ipynb b/example notebooks/edm tutorial.ipynb → tutorial notebooks/edm tutorial.ipynb
diff --git a/...ple notebooks/simple diffusion flax.ipynb → ...ial notebooks/simple diffusion flax.ipynb b/...ple notebooks/simple diffusion flax.ipynb → ...ial notebooks/simple diffusion flax.ipynb
diff --git a/tutorial notebooks/text2image with CFG.ipynb b/tutorial notebooks/text2image with CFG.ipynb