Fixed args to forward

video2world_hf.py

@@ -77,7 +77,7 @@ class DiffusionVideo2World(PreTrainedModel):
             num_input_frames=cfg.num_input_frames,
         )
 
-    def forward(self):
+    def forward(self, prompt, input_image_or_video_path):
         cfg = self.config
 
         # Handle multiple prompts if prompt file is provided
@@ -86,7 +86,7 @@ class DiffusionVideo2World(PreTrainedModel):
             prompts = read_prompts_from_file(cfg.batch_input_path)
         else:
             # Single prompt case
-            prompts = [{"prompt": cfg.prompt, "visual_input": cfg.input_image_or_video_path}]
+            prompts = [{"prompt": prompt, "visual_input": input_image_or_video_path}]
 
         os.makedirs(cfg.video_save_folder, exist_ok=True)
         for i, input_dict in enumerate(prompts):