update

app.py

@@ -221,7 +221,7 @@ def generation_step(runner, text_embeds_dict, cond_latents):
     return samples
 
 @spaces.GPU(duration=120)
-def generation_loop(video_path='./test_videos', output_dir='./results', seed=666, batch_size=1, cfg_scale=1.0, cfg_rescale=0.0, sample_steps=1, res_h=1280, res_w=720, sp_size=1):
+def generation_loop(video_path='./test_videos', seed=666, batch_size=1, cfg_scale=1.0, cfg_rescale=0.0, sample_steps=1, res_h=1280, res_w=720, sp_size=1):
     runner = configure_runner(1)
     output_dir = 'output/out.mp4'
     def _build_pos_and_neg_prompt():
@@ -386,6 +386,7 @@ def generation_loop(video_path='./test_videos', output_dir='./results', seed=666
             sample = sample.to(torch.uint8).numpy()
 
         gc.collect()
+        torch.cuda.empty_cache()
         return sample, output_dir
 
 
@@ -461,9 +462,9 @@ demo = gr.Interface(
 	description=description,
 	article=article,
 	examples=[
-		['./01.mp4', 4, 0.5, 42, "512", 200, "adain"],
-		['./02.mp4', 4, 0.5, 42, "512", 200, "adain"],
-		['./03.mp4', 4, 0.5, 42, "512", 200, "adain"],
+		['./01.mp4', 4],
+		['./02.mp4', 4],
+		['./03.mp4', 4],
 		]
 	)