Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -301,15 +301,18 @@ def generate_video(
|
|
301 |
num_skip_start_steps,
|
302 |
enable_riflex,
|
303 |
riflex_k,
|
304 |
-
progress=gr.Progress(
|
305 |
):
|
306 |
global pipeline, wav2vec_processor, wav2vec_model, config
|
307 |
|
308 |
progress(0, desc="Starting video generation...")
|
309 |
|
310 |
-
if image_path is None:
|
311 |
-
|
312 |
-
if
|
|
|
|
|
|
|
313 |
|
314 |
device = pipeline.device
|
315 |
|
@@ -391,7 +394,8 @@ def generate_video(
|
|
391 |
int((current_partial_length - 1) // pipeline.vae.config.temporal_compression_ratio * pipeline.vae.config.temporal_compression_ratio) + 1
|
392 |
if current_partial_length > 1 else 1
|
393 |
)
|
394 |
-
if current_partial_length <= 0:
|
|
|
395 |
|
396 |
input_video, input_video_mask, clip_image = get_image_to_video_latent3(
|
397 |
ref_img_for_loop, None, video_length=current_partial_length,
|
@@ -574,7 +578,7 @@ def create_demo():
|
|
574 |
use_un_ip_mask = gr.Checkbox(label="Use Un-IP Mask", value=False, info="Inverts the inpainting mask.")
|
575 |
|
576 |
with gr.Row():
|
577 |
-
|
578 |
gr.Markdown("### Negative Guidance (Advanced CFG)")
|
579 |
neg_scale = gr.Slider(label="Negative Scale", minimum=1.0, maximum=5.0, value=1.5, step=0.1, info="Strength of negative prompt in early steps.")
|
580 |
neg_steps = gr.Slider(label="Negative Steps", minimum=0, maximum=10, value=2, step=1, info="How many initial steps to apply the negative scale.")
|
|
|
301 |
num_skip_start_steps,
|
302 |
enable_riflex,
|
303 |
riflex_k,
|
304 |
+
progress=gr.Progress(track_tqdm=True)
|
305 |
):
|
306 |
global pipeline, wav2vec_processor, wav2vec_model, config
|
307 |
|
308 |
progress(0, desc="Starting video generation...")
|
309 |
|
310 |
+
if image_path is None:
|
311 |
+
raise gr.Error("Please upload an image")
|
312 |
+
if audio_path is None:
|
313 |
+
raise gr.Error("Please upload an audio file")
|
314 |
+
if not models_ready or pipeline is None:
|
315 |
+
raise gr.Error("Models not initialized. Please restart the space.")
|
316 |
|
317 |
device = pipeline.device
|
318 |
|
|
|
394 |
int((current_partial_length - 1) // pipeline.vae.config.temporal_compression_ratio * pipeline.vae.config.temporal_compression_ratio) + 1
|
395 |
if current_partial_length > 1 else 1
|
396 |
)
|
397 |
+
if current_partial_length <= 0:
|
398 |
+
break
|
399 |
|
400 |
input_video, input_video_mask, clip_image = get_image_to_video_latent3(
|
401 |
ref_img_for_loop, None, video_length=current_partial_length,
|
|
|
578 |
use_un_ip_mask = gr.Checkbox(label="Use Un-IP Mask", value=False, info="Inverts the inpainting mask.")
|
579 |
|
580 |
with gr.Row():
|
581 |
+
with gr.Column():
|
582 |
gr.Markdown("### Negative Guidance (Advanced CFG)")
|
583 |
neg_scale = gr.Slider(label="Negative Scale", minimum=1.0, maximum=5.0, value=1.5, step=0.1, info="Strength of negative prompt in early steps.")
|
584 |
neg_steps = gr.Slider(label="Negative Steps", minimum=0, maximum=10, value=2, step=1, info="How many initial steps to apply the negative scale.")
|