ValueError: xformers is not available. Make sure it is installed correctly

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

ValueError: xformers is not available. Make sure it is installed correctly #11

Closed Dylandtt closed 3 months ago

Dylandtt commented 4 months ago

训练时可以不安装xformers吗

flymin commented 4 months ago

Yes, you can disable it here or through command line like our demo, but it leads to much more gpu memory cost. You may use fewer bs or lower resolutions if OOM. Besides, we also support gradient checkpointing. Choose your own configurations wisely.