search

shikras / shikra

Other
710 stars 44 forks source link

which torch version used? #6

Closed JiaweiZhao-git closed 11 months ago

JiaweiZhao-git commented 1 year ago

the new version is 2.0.1, I wonder this project use 1. or 2.?

kq-chen commented 11 months ago

The torch and torchvision utilized in our environment are torch-1.13.1+cu117 and torchvision-0.14.1+cu117