Fabrice-TIERCELIN commited on
Commit
3d932ba
1 Parent(s): 455c2e5

This PR allows the user to automatically randomize the seed

Browse files
Files changed (1) hide show
  1. app.py +14 -9
app.py CHANGED
@@ -134,6 +134,7 @@ class FoleyController:
134
  sampler_dropdown,
135
  sample_step_slider,
136
  cfg_scale_slider,
 
137
  seed_textbox,
138
  ):
139
  device = 'cuda'
@@ -151,9 +152,11 @@ class FoleyController:
151
  # if not self.loaded:
152
  # raise gr.Error("Error with loading model")
153
  generator = torch.Generator()
154
- if seed_textbox != "":
155
- torch.manual_seed(int(seed_textbox))
156
- generator.manual_seed(int(seed_textbox))
 
 
157
  max_frame_nums = 150
158
  frames, duration = read_frames_with_moviepy(input_video, max_frame_nums=max_frame_nums)
159
  if duration >= 10:
@@ -271,7 +274,8 @@ with gr.Blocks(css=css) as demo:
271
  cfg_scale_slider = gr.Slider(label="CFG Scale", value=7.5, minimum=0, maximum=20)
272
 
273
  with gr.Row():
274
- seed_textbox = gr.Textbox(label="Seed", value=42)
 
275
  seed_button = gr.Button(value="\U0001f3b2", elem_classes="toolbutton")
276
  seed_button.click(fn=lambda x: random.randint(1, 1e8), outputs=[seed_textbox], queue=False)
277
 
@@ -302,6 +306,7 @@ with gr.Blocks(css=css) as demo:
302
  sampler_dropdown,
303
  sample_step_slider,
304
  cfg_scale_slider,
 
305
  seed_textbox,
306
  ],
307
  outputs=[result_video],
@@ -315,12 +320,12 @@ with gr.Blocks(css=css) as demo:
315
  # ['examples/videos/2.mp4', '', '', 1.0, 'DDIM', 25, 7.5, 16520432],
316
  # ],
317
  examples=[
318
- ['examples/input/case1.mp4', '', '', 1.0, 0.2, 'DDIM', 25, 7.5, 33817921],
319
- ['examples/input/case3.mp4', '', '', 1.0, 0.2,'DDIM', 25, 7.5, 94667578],
320
- ['examples/input/case5.mp4', '', '', 0.75, 0.2,'DDIM', 25, 7.5, 92890876],
321
- ['examples/input/case6.mp4', '', '', 1.0, 0.2, 'DDIM', 25, 7.5, 77015909],
322
  ],
323
- inputs=[init_img,prompt_textbox,negative_prompt_textbox,ip_adapter_scale,temporal_scale,sampler_dropdown,sample_step_slider,cfg_scale_slider,seed_textbox],
324
  cache_examples=False,
325
  outputs=[result_video],
326
  fn=controller.foley,
 
134
  sampler_dropdown,
135
  sample_step_slider,
136
  cfg_scale_slider,
137
+ randomize_seed,
138
  seed_textbox,
139
  ):
140
  device = 'cuda'
 
152
  # if not self.loaded:
153
  # raise gr.Error("Error with loading model")
154
  generator = torch.Generator()
155
+ if is_randomize_seed:
156
+ seed_textbox = random.randint(1, 1e8)
157
+ if seed_textbox is not None:
158
+ torch.manual_seed(seed_textbox)
159
+ generator.manual_seed(seed_textbox)
160
  max_frame_nums = 150
161
  frames, duration = read_frames_with_moviepy(input_video, max_frame_nums=max_frame_nums)
162
  if duration >= 10:
 
274
  cfg_scale_slider = gr.Slider(label="CFG Scale", value=7.5, minimum=0, maximum=20)
275
 
276
  with gr.Row():
277
+ randomize_seed = gr.Checkbox(label = "\U0001F3B2 Randomize seed", value = True, info = "If checked, result is always different")
278
+ seed_textbox = gr.Slider(label="Seed", minimum=1, maximum=1e8 - 1, step=1, randomize=True)
279
  seed_button = gr.Button(value="\U0001f3b2", elem_classes="toolbutton")
280
  seed_button.click(fn=lambda x: random.randint(1, 1e8), outputs=[seed_textbox], queue=False)
281
 
 
306
  sampler_dropdown,
307
  sample_step_slider,
308
  cfg_scale_slider,
309
+ randomize_seed,
310
  seed_textbox,
311
  ],
312
  outputs=[result_video],
 
320
  # ['examples/videos/2.mp4', '', '', 1.0, 'DDIM', 25, 7.5, 16520432],
321
  # ],
322
  examples=[
323
+ ['examples/input/case1.mp4', '', '', 1.0, 0.2, 'DDIM', 25, 7.5, False, 33817921],
324
+ ['examples/input/case3.mp4', '', '', 1.0, 0.2,'DDIM', 25, 7.5, False, 94667578],
325
+ ['examples/input/case5.mp4', '', '', 0.75, 0.2,'DDIM', 25, 7.5, False, 92890876],
326
+ ['examples/input/case6.mp4', '', '', 1.0, 0.2, 'DDIM', 25, 7.5, False, 77015909],
327
  ],
328
+ inputs=[init_img,prompt_textbox,negative_prompt_textbox,ip_adapter_scale,temporal_scale,sampler_dropdown,sample_step_slider,cfg_scale_slider,randomize_seed,seed_textbox],
329
  cache_examples=False,
330
  outputs=[result_video],
331
  fn=controller.foley,