From 57aa8c24199f169eb57f04ff5ea33182e3e01701 Mon Sep 17 00:00:00 2001 From: Ashish Kumar Singh Date: Tue, 16 Jul 2024 22:55:36 +0530 Subject: [PATCH] feat: some more changes --- Diffusion flax linen.ipynb | 38 ++++++++++++++++++++++++++++++++++++++ README.md | 36 +++++++++++++++++++++--------------- 2 files changed, 59 insertions(+), 15 deletions(-) diff --git a/Diffusion flax linen.ipynb b/Diffusion flax linen.ipynb index fb822b6..efee46d 100644 --- a/Diffusion flax linen.ipynb +++ b/Diffusion flax linen.ipynb @@ -1840,6 +1840,44 @@ "textEncoderModel, textTokenizer = defaultTextEncodeModel()" ] }, + { + "cell_type": "code", + "execution_count": 35, + "metadata": {}, + "outputs": [ + { + "data": { + "text/plain": [ + "'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'" + ] + }, + "execution_count": 35, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "prompts = [\n", + " 'water tulip',\n", + " 'a water lily',\n", + " 'a water lily', \n", + " 'a water lily', \n", + " 'a photo of a marigold', \n", + " 'a water lily',\n", + " 'a water lily',\n", + " 'a photo of a lotus',\n", + " 'a photo of a lotus',\n", + " 'a photo of a lotus',\n", + " 'a photo of a rose',\n", + " 'a photo of a rose',\n", + " 'a photo of a rose',\n", + " 'a photo of a rose',\n", + " 'a photo of a rose',\n", + " ]\n", + "\n", + "', '.join(prompts)" + ] + }, { "cell_type": "code", "execution_count": 17, diff --git a/README.md b/README.md index 2d25f05..dbd9c53 100644 --- a/README.md +++ b/README.md @@ -203,21 +203,17 @@ plotImages(samples, dpi=300) ### 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' +`'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'` + +**Params**: +`Dataset: oxford_flowers102` +`Batch size: 16` +`Image Size: 128` +`Training Epochs: 1000` +`Steps per epoch: 511` + +`Training Noise Schedule: EDMNoiseScheduler` +`Inference Noise Schedule: KarrasEDMPredictor` ![EulerA with CFG](images/text2img%20euler%20ancestral%201.png) @@ -225,6 +221,16 @@ Images generated by the following prompts using classifier free guidance with gu 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'` +**Params**: +`Dataset: oxford_flowers102` +`Batch size: 16` +`Image Size: 128` +`Training Epochs: 1000` +`Steps per epoch: 511` + +`Training Noise Schedule: EDMNoiseScheduler` +`Inference Noise Schedule: KarrasEDMPredictor` + ![EulerA with CFG](images/text2img%20euler%20ancestral%202.png) ### Images generated by DDPM Sampler in 1000 steps [Unconditional]