random seed

diffuzers/__init__.py

@@ -5,4 +5,4 @@
 
 logger.configure(handlers=[dict(sink=sys.stderr, format="> <level>{level:<7} {message}</level>")])
 
-__version__ = "0.2.8"
+__version__ = "0.2.9"

diffuzers/api/main.py

@@ -68,6 +68,7 @@ async def text2img(params: Text2ImgParams) -> ImgResponse:
     logger.info(f"Params: {params}")
     if app.state.x2img_model is None:
         return {"error": "x2img model is not loaded"}
+
     images, _ = app.state.x2img_model.text2img_generate(
         params.prompt,
         num_images=params.num_images,

diffuzers/inpainting.py

@@ -1,5 +1,6 @@
 import gc
 import json
+import random
 from dataclasses import dataclass
 from io import BytesIO
 from typing import Optional
@@ -67,6 +68,11 @@ def _set_scheduler(self, scheduler_name):
     def generate_image(
         self, prompt, negative_prompt, image, mask, guidance_scale, scheduler, steps, seed, height, width, num_images
     ):
+
+        if seed == -1:
+            # generate random seed
+            seed = random.randint(0, 999999)
+
         self._set_scheduler(scheduler)
         logger.info(self.pipeline.scheduler)
 
@@ -176,11 +182,12 @@ def app(self):
         seed = st.sidebar.number_input(
             "Seed",
             value=42,
-            min_value=1,
+            min_value=-1,
             max_value=999999,
             step=1,
             help="Random seed. Change for different results using same parameters.",
         )
+
         if uploaded_file is not None:
             with col2:
                 drawing_mode = st.selectbox(

diffuzers/x2image.py

@@ -2,6 +2,7 @@
 import gc
 import json
 import os
+import random
 import tempfile
 from dataclasses import dataclass
 from io import BytesIO
@@ -163,6 +164,11 @@ def _postgen(self, metadata, output_images, pipeline_name):
     def text2img_generate(
         self, prompt, negative_prompt, scheduler, image_size, num_images, guidance_scale, steps, seed
     ):
+
+        if seed == -1:
+            # generate random seed
+            seed = random.randint(0, 999999)
+
         generator, num_images = self._pregen(
             pipeline_name="text2img",
             scheduler=scheduler,
@@ -200,6 +206,11 @@ def text2img_generate(
     def img2img_generate(
         self, prompt, image, strength, negative_prompt, scheduler, num_images, guidance_scale, steps, seed
     ):
+
+        if seed == -1:
+            # generate random seed
+            seed = random.randint(0, 999999)
+
         generator, num_images = self._pregen(
             pipeline_name="img2img",
             scheduler=scheduler,
@@ -330,11 +341,12 @@ def app(self):
         seed = st.sidebar.number_input(
             "Seed",
             value=42,
-            min_value=1,
+            min_value=-1,
             max_value=999999,
             step=1,
             help="Random seed. Change for different results using same parameters.",
         )
+
         sub_col, download_col = st.columns(2)
         with sub_col:
             submit = st.button("Generate")