Fixed some arg defaults

futureworld_hf.py

@@ -59,10 +59,10 @@ class AutoregressiveFutureWorld(PreTrainedModel):
 
     # modifed from text2world.py demo function
     def forward(self, input_type,
-                batch_input_path,
-                input_image_or_video_path,
-                data_resolution,
-                num_input_frames):
+                batch_input_path = None,
+                input_image_or_video_path = None,
+                data_resolution = None,
+                num_input_frames = 9):
         input_videos = load_vision_input(
             input_type=input_type,
             batch_input_path=batch_input_path,

test_hf.py

@@ -15,5 +15,10 @@ model = AutoModel.from_pretrained("EthanZyh/DiffusionText2WorldGeneration",
 #model(prompt)
 
 model = AutoModel.from_pretrained("sheldonl/AutoregressiveFutureWorld",
+                                  ar_model_dir = "Cosmos-1.0-Autoregressive-4B",
                                   trust_remote_code = True)
 
+model(input_type = "video",
+      input_image_or_video_path = "../Cosmos/cosmos1/models/autoregressive/assets/v1p0/input.mp4",
+      )
+