This needs to be installed according to your CUDA version. Please refer to the official PyTorch installation guide. The following example uses CUDA 12: pip install ...
Generative diffusion models for end-to-end autonomous driving often suffer from mode collapse, tending to generate conservative and homogeneous behaviors. While DiffusionDrive employs predefined ...