Linoy Tsaban commited on
Commit
3bcf222
1 Parent(s): 0479d0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -107,7 +107,12 @@ with gr.Blocks(css='style.css') as demo:
107
  cfg_scale_src = 3.5,
108
  cfg_scale_tar = 15,
109
  skip=36,
110
- seed = 0,):
 
 
 
 
 
111
  torch.manual_seed(seed)
112
  # offsets=(0,0,0,0)
113
  x0 = load_512(input_image, device=device)
@@ -151,6 +156,7 @@ with gr.Blocks(css='style.css') as demo:
151
  skip = gr.Slider(minimum=0, maximum=40, value=36, precision=0, label="Skip Steps", interactive=True)
152
  cfg_scale_tar = gr.Slider(minimum=7, maximum=18,value=15, label=f"Target Guidance Scale", interactive=True)
153
  seed = gr.Number(value=0, precision=0, label="Seed", interactive=True)
 
154
 
155
 
156
  edit_button.click(
 
107
  cfg_scale_src = 3.5,
108
  cfg_scale_tar = 15,
109
  skip=36,
110
+ seed = 0,
111
+ randomized_seed = True):
112
+
113
+ if randomized_seed:
114
+ seed = random.randint(0, np.iinfo(np.int32).max)
115
+
116
  torch.manual_seed(seed)
117
  # offsets=(0,0,0,0)
118
  x0 = load_512(input_image, device=device)
 
156
  skip = gr.Slider(minimum=0, maximum=40, value=36, precision=0, label="Skip Steps", interactive=True)
157
  cfg_scale_tar = gr.Slider(minimum=7, maximum=18,value=15, label=f"Target Guidance Scale", interactive=True)
158
  seed = gr.Number(value=0, precision=0, label="Seed", interactive=True)
159
+ randomize_seed = gr.Checkbox(label='Randomize seed', value=True)
160
 
161
 
162
  edit_button.click(