torch==2.1.0 torchaudio==2.1.0 git+https://github.com/facebookresearch/audiocraft.git@magnet_xformers_0_0_22_fix xformers --index-url https://download.pytorch.org/whl/cu118