Dependencies version requirements

[kianpu34593]

Hi,

In the dependency installation section, it is said that pip install torch==2.0.1. Is Pytorch v2.0.1 required?
I installed Pytorch v2.2.2 with cuda11.8. It doesn't have any problem doing the inferencing.

Just wondering if this requirement can be relaxed if I plan to do training in the future.

Best,
Kian

[lantunes]

Hello,

Thanks for asking this question. The documentation specifies PyTorch 2.0.1 because that's the version with which we've developed and thoroughly tested the code. However, as you've successfully run inference with PyTorch v2.2.2, it suggests that later versions may also be compatible. If you plan to use PyTorch v2.2.2 for training as well, it could work, but please be aware that we haven't formally tested this version.

Best regards,
Luis