Upload 29 files

diffusion_webui/__init__.py

@@ -1 +1 @@
-__version__ = "1.8.0"
+__version__ = "2.0.1"

diffusion_webui/diffusion_models/stable_diffusion/text2img_app.py

@@ -3,7 +3,10 @@ import torch
 from diffusers import StableDiffusionPipeline
 
 from diffusion_webui.utils.model_list import stable_model_list
-from diffusion_webui.utils.scheduler_list import get_scheduler_list
+from diffusion_webui.utils.scheduler_list import (
+    SCHEDULER_LIST,
+    get_scheduler_list,
+)
 
 
 class StableDiffusionText2ImageGenerator:
@@ -111,14 +114,8 @@ class StableDiffusionText2ImageGenerator:
                             with gr.Column():
 
                                 text2image_scheduler = gr.Dropdown(
-                                    choices=[
-                                        "DDIM",
-                                        "EulerA",
-                                        "Euler",
-                                        "LMS",
-                                        "Heun",
-                                    ],
-                                    value="DDIM",
+                                    choices=SCHEDULER_LIST,
+                                    value=SCHEDULER_LIST[0],
                                     label="Scheduler",
                                 )
 

diffusion_webui/upscaler_models/codeformer_upscaler.py

@@ -11,14 +11,14 @@ class CodeformerUpscalerGenerator:
         upscale: int,
         codeformer_fidelity: int,
     ):
-        
+
         pipe = inference_app(
-                image=image_path,
-                background_enhance=background_enhance,
-                face_upsample=face_upsample,
-                upscale=upscale,
-                codeformer_fidelity=codeformer_fidelity,
-            )
+            image=image_path,
+            background_enhance=background_enhance,
+            face_upsample=face_upsample,
+            upscale=upscale,
+            codeformer_fidelity=codeformer_fidelity,
+        )
 
         return [pipe]
 

diffusion_webui/utils/data_utils.py

@@ -0,0 +1,12 @@
+from PIL import Image
+
+
+def image_grid(imgs, rows, cols):
+    assert len(imgs) == rows * cols
+
+    w, h = imgs[0].size
+    grid = Image.new("RGB", size=(cols * w, rows * h))
+
+    for i, img in enumerate(imgs):
+        grid.paste(img, box=(i % cols * w, i // cols * h))
+    return grid

diffusion_webui/utils/model_list.py

@@ -2,6 +2,12 @@ stable_model_list = [
     "runwayml/stable-diffusion-v1-5",
     "stabilityai/stable-diffusion-2-1",
     "prompthero/openjourney-v4",
+    "wavymulder/Analog-Diffusion",
+    "dreamlike-art/dreamlike-diffusion-1.0",
+    "gsdf/Counterfeit-V2.5",
+    "dreamlike-art/dreamlike-photoreal-2.0"
+    
+    
 ]
 
 controlnet_canny_model_list = [