Spaces:
Runtime error
Runtime error
This PR allows the user to automatically randomize the seed
#2
by
Fabrice-TIERCELIN
- opened
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
|
| 155 |
-
|
| 156 |
-
|
|
|
|
|
|
|
| 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 |
-
|
|
|
|
| 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,
|