Commit graph

2 commits

Author SHA1 Message Date
xzeng 4d6af1d8b9 fix tensor device issue
https://github.com/nv-tlabs/LION/issues/31#issue-1627736930
the original code index cpu tensor with cuda tensor (works in torch 1.10.2),
may fail in other torch version?
2023-03-16 12:44:57 -04:00
xzeng 1d24a7879d init 2023-01-23 00:14:49 -05:00