wrong grid size

[monicamini]

when running demo_match.py, it will show:

(romatch) root@autodl-container-09dd4b98a3-18c5ddd8:~/RoMa# python demo_match.py
2025-12-18 21:55:36.275 | INFO | romatch.models.model_zoo.roma_models:roma_model:61 - Using coarse resolution (560, 560), and upsample res (864, 1152)
Traceback (most recent call last):
File "/root/RoMa/demo_match.py", line 41, in
im2_transfer_rgb = F.grid_sample(
^^^^^^^^^^^^^^
File "/root/RoMa/.venv/lib/python3.12/site-packages/torch/nn/functional.py", line 5104, in grid_sample
return torch.grid_sampler(input, grid, mode_enum, padding_mode_enum, align_corners)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: grid_sampler(): expected grid to have size 2 in last dimension, but got grid with sizes [1, 1, 864, 2302, 4]

[Catsgomeowmeow]

I think you meant to post this error in RoMa V1 not v2 ? "romatch.models.model_zoo.roma_models:roma_model" is the file structure from v1