cure-lab / MagicDrive

[ICLR24] Official implementation of the paper “MagicDrive: Street View Generation with Diverse 3D Geometry Control”
https://gaoruiyuan.com/magicdrive/
GNU Affero General Public License v3.0
419 stars 22 forks source link

torch version and xformers install #24

Closed Leedonus closed 2 months ago

Leedonus commented 3 months ago

Thanks for sharing this work! I would like to ask if the torch version needs to be 1.12.0+ when training, since xformer relies on 1.12.0+. However, torch version in README is 1.10.2. 2

flymin commented 3 months ago

No, as said in our doc, we used torch 1.10.2.

BTW, if you have problems with xformers, please ignore the package. It is totally optional as said in the readme.