Spaces:
Running
on
Zero
Running
on
Zero
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,
|