ControlNet-v1-1

Paused

App Files Files Community

hysts HF staff commited on Jul 7, 2023

Commit

06d6d46

•

1 Parent(s): d5479f6

Update

Browse files

Files changed (14) hide show

app_canny.py +3 -4
app_depth.py +3 -4
app_ip2p.py +3 -4
app_lineart.py +3 -4
app_mlsd.py +3 -4
app_normal.py +3 -4
app_openpose.py +3 -4
app_scribble.py +3 -4
app_scribble_interactive.py +3 -4
app_segmentation.py +3 -4
app_shuffle.py +3 -4
app_softedge.py +3 -4
settings.py +4 -0
utils.py +3 -1

app_canny.py CHANGED Viewed

@@ -3,7 +3,7 @@
 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
-                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES)
 from utils import randomize_seed_fn
@@ -50,10 +50,9 @@ def create_demo(process):
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
-                                     maximum=1000000,
                                      step=1,
-                                     value=0,
-                                     randomize=True)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
+                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
 from utils import randomize_seed_fn
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
+                                     maximum=MAX_SEED,
                                      step=1,
+                                     value=0)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

app_depth.py CHANGED Viewed

@@ -3,7 +3,7 @@
 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
-                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES)
 from utils import randomize_seed_fn
@@ -49,10 +49,9 @@ def create_demo(process):
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
-                                     maximum=1000000,
                                      step=1,
-                                     value=0,
-                                     randomize=True)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
+                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
 from utils import randomize_seed_fn
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
+                                     maximum=MAX_SEED,
                                      step=1,
+                                     value=0)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

app_ip2p.py CHANGED Viewed

@@ -3,7 +3,7 @@
 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
-                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES)
 from utils import randomize_seed_fn
@@ -38,10 +38,9 @@ def create_demo(process):
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
-                                     maximum=1000000,
                                      step=1,
-                                     value=0,
-                                     randomize=True)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
+                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
 from utils import randomize_seed_fn
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
+                                     maximum=MAX_SEED,
                                      step=1,
+                                     value=0)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

app_lineart.py CHANGED Viewed

@@ -3,7 +3,7 @@
 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
-                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES)
 from utils import randomize_seed_fn
@@ -58,10 +58,9 @@ def create_demo(process):
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
-                                     maximum=1000000,
                                      step=1,
-                                     value=0,
-                                     randomize=True)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
+                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
 from utils import randomize_seed_fn
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
+                                     maximum=MAX_SEED,
                                      step=1,
+                                     value=0)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

app_mlsd.py CHANGED Viewed

@@ -3,7 +3,7 @@
 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
-                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES)
 from utils import randomize_seed_fn
@@ -56,10 +56,9 @@ def create_demo(process):
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
-                                     maximum=1000000,
                                      step=1,
-                                     value=0,
-                                     randomize=True)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
+                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
 from utils import randomize_seed_fn
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
+                                     maximum=MAX_SEED,
                                      step=1,
+                                     value=0)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

app_normal.py CHANGED Viewed

@@ -3,7 +3,7 @@
 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
-                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES)
 from utils import randomize_seed_fn
@@ -48,10 +48,9 @@ def create_demo(process):
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
-                                     maximum=1000000,
                                      step=1,
-                                     value=0,
-                                     randomize=True)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
+                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
 from utils import randomize_seed_fn
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
+                                     maximum=MAX_SEED,
                                      step=1,
+                                     value=0)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

app_openpose.py CHANGED Viewed

@@ -3,7 +3,7 @@
 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
-                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES)
 from utils import randomize_seed_fn
@@ -48,10 +48,9 @@ def create_demo(process):
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
-                                     maximum=1000000,
                                      step=1,
-                                     value=0,
-                                     randomize=True)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
+                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
 from utils import randomize_seed_fn
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
+                                     maximum=MAX_SEED,
                                      step=1,
+                                     value=0)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

app_scribble.py CHANGED Viewed

@@ -3,7 +3,7 @@
 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
-                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES)
 from utils import randomize_seed_fn
@@ -49,10 +49,9 @@ def create_demo(process):
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
-                                     maximum=1000000,
                                      step=1,
-                                     value=0,
-                                     randomize=True)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
+                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
 from utils import randomize_seed_fn
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
+                                     maximum=MAX_SEED,
                                      step=1,
+                                     value=0)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

app_scribble_interactive.py CHANGED Viewed

@@ -4,7 +4,7 @@ import gradio as gr
 import numpy as np
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
-                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES)
 from utils import randomize_seed_fn
@@ -54,10 +54,9 @@ def create_demo(process):
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
-                                     maximum=1000000,
                                      step=1,
-                                     value=0,
-                                     randomize=True)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

 import numpy as np
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
+                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
 from utils import randomize_seed_fn
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
+                                     maximum=MAX_SEED,
                                      step=1,
+                                     value=0)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

app_segmentation.py CHANGED Viewed

@@ -3,7 +3,7 @@
 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
-                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES)
 from utils import randomize_seed_fn
@@ -48,10 +48,9 @@ def create_demo(process):
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
-                                     maximum=1000000,
                                      step=1,
-                                     value=0,
-                                     randomize=True)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
+                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
 from utils import randomize_seed_fn
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
+                                     maximum=MAX_SEED,
                                      step=1,
+                                     value=0)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

app_shuffle.py CHANGED Viewed

@@ -3,7 +3,7 @@
 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
-                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES)
 from utils import randomize_seed_fn
@@ -43,10 +43,9 @@ def create_demo(process):
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
-                                     maximum=1000000,
                                      step=1,
-                                     value=0,
-                                     randomize=True)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
+                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
 from utils import randomize_seed_fn
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
+                                     maximum=MAX_SEED,
                                      step=1,
+                                     value=0)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

app_softedge.py CHANGED Viewed

@@ -3,7 +3,7 @@
 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
-                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES)
 from utils import randomize_seed_fn
@@ -54,10 +54,9 @@ def create_demo(process):
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
-                                     maximum=1000000,
                                      step=1,
-                                     value=0,
-                                     randomize=True)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

 import gradio as gr
 from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
+                      MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
 from utils import randomize_seed_fn
                                                step=0.1)
                     seed = gr.Slider(label='Seed',
                                      minimum=0,
+                                     maximum=MAX_SEED,
                                      step=1,
+                                     value=0)
                     randomize_seed = gr.Checkbox(label='Randomize seed',
                                                  value=True)
                     a_prompt = gr.Textbox(

settings.py CHANGED Viewed

@@ -1,5 +1,7 @@
 import os
 DEFAULT_MODEL_ID = os.getenv('DEFAULT_MODEL_ID',
                              'runwayml/stable-diffusion-v1-5')
@@ -11,3 +13,5 @@ DEFAULT_IMAGE_RESOLUTION = int(os.getenv('DEFAULT_IMAGE_RESOLUTION', '768'))
 ALLOW_CHANGING_BASE_MODEL = os.getenv('SPACE_ID') != 'hysts/ControlNet-v1-1'
 SHOW_DUPLICATE_BUTTON = os.getenv('SHOW_DUPLICATE_BUTTON') == '1'

 import os
+import numpy as np
 DEFAULT_MODEL_ID = os.getenv('DEFAULT_MODEL_ID',
                              'runwayml/stable-diffusion-v1-5')
 ALLOW_CHANGING_BASE_MODEL = os.getenv('SPACE_ID') != 'hysts/ControlNet-v1-1'
 SHOW_DUPLICATE_BUTTON = os.getenv('SHOW_DUPLICATE_BUTTON') == '1'
+MAX_SEED = np.iinfo(np.int32).max

utils.py CHANGED Viewed

@@ -1,7 +1,9 @@
 import random
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
-        seed = random.randint(0, 1000000)
     return seed

 import random
+from settings import MAX_SEED
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
     return seed