Update worker_runpod.py

worker_runpod.py

@@ -134,16 +134,7 @@ def generate(input):
     input_video, input_video_mask, clip_image = get_image_to_video_latent(downloaded_image_path, validation_image_end, video_length=video_length, sample_size=sample_size)
         
     with torch.no_grad():
-    sample = pipeline(prompt=prompt,
-                num_frames=video_length,
-                negative_prompt=negative_prompt,
-                height=sample_size[0],
-                width=sample_size[1],
-                generator=generator,
-                guidance_scale=guidance_scale,
-                num_inference_steps=num_inference_steps,
-                video=input_video,
-                mask_video=input_video_mask).videos        
+        sample = pipeline(prompt=prompt,num_frames=video_length,negative_prompt=negative_prompt,height=sample_size[0],width=sample_size[1],generator=generator,guidance_scale=guidance_scale,num_inference_steps=num_inference_steps,video=input_video,mask_video=input_video_mask).videos        
     
     if not os.path.exists(save_path):
         os.makedirs(save_path, exist_ok=True)