Added data resolution default val
Browse files- futureworld_hf.py +1 -1
futureworld_hf.py
CHANGED
|
@@ -61,7 +61,7 @@ class AutoregressiveFutureWorld(PreTrainedModel):
|
|
| 61 |
def forward(self, input_type,
|
| 62 |
batch_input_path = None,
|
| 63 |
input_image_or_video_path = None,
|
| 64 |
-
data_resolution =
|
| 65 |
num_input_frames = 9):
|
| 66 |
input_videos = load_vision_input(
|
| 67 |
input_type=input_type,
|
|
|
|
| 61 |
def forward(self, input_type,
|
| 62 |
batch_input_path = None,
|
| 63 |
input_image_or_video_path = None,
|
| 64 |
+
data_resolution = [640, 1024],
|
| 65 |
num_input_frames = 9):
|
| 66 |
input_videos = load_vision_input(
|
| 67 |
input_type=input_type,
|