feat: some more changes

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,

README.md

@@ -203,28 +203,34 @@ 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)
 
 ### 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'`
 
+**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]