kyegomez / zeta

Build high-performance AI models with modular building blocks
https://zeta.apac.ai
Apache License 2.0
320 stars 28 forks source link

[BUG] Torchvision requirement is pinned to 0.18.0 #232

Open ClementMaliet opened 3 weeks ago

ClementMaliet commented 3 weeks ago

Describe the bug Hello,

Although most requirements are lax, the torchvision dependency version is pinned to 0.18.0 which in turns prevents the use of the new PyTorch 2.3.1 version. I need to use the 2.3.1 version of PyTorch to conform with other dependencies that I have but zeta puts me into a dependency hell.

Is there any reason to put such a hard pin on torchvision ?

Upvote & Fund

Fund with Polar

github-actions[bot] commented 3 weeks ago

Hello there, thank you for opening an Issue ! 🙏🏻 The team was notified and they will get back to you asap.

kyegomez commented 3 weeks ago

@ClementMaliet we can change it!