Questions about VideoChat2_HD

Hi, thanks for your update of VideoChat2_HD! When trying the newly-released code, I got some questions:

• The MetaLoader_rs class in "train_it_ds.py" seems to be missing.
• So I still used "train_it.py", but got the following error. I'm not sure whether it could be solved by using MetaLoader_rs.

RuntimeError: stack expects each tensor to be equal size, but got [8, 3, 224, 448] at entry 0 and [8, 3, 448, 672] at entry 1

• Then I changed the batch_size to 1 and solved the previous error. But it seems the load_and_transform_media_data_image function does not have dynamic_config setting, which is passed to it in "it_dataset_mistral.py". I created a pull request to modify this part.
• Is there any place to find the newly added dataset for VideoChat2_HD? I suppose the datasets are important to improve model performances.

Start by comparing train_it_ds.py and train_it.py, then inspect it_dataset_mistral.py where dynamic_config is passed to load_and_transform_media_data_image. Reproduce the reported batching error and check where the VideoChat2_HD dataset is documented or loaded; done means the missing class, batch-size failure, and dynamic_config question have clear resolutions.