Update requirements.txt

requirements.txt

@@ -1,26 +1,26 @@
-accelerate==1.0.1
-bitsandbytes==0.44.1
-deepspeed
-diffusers==0.19.3
-einops
-huggingface-hub
-imageio[ffmpeg]
-invisible_watermark
-omegaconf
-peft
-plyfile
-PyMCubes
-pytorch-lightning==2.1.2
-rembg
-sentencepiece
-tensorboard
-torch==2.1.0
-torchvision==0.16.0
-torchaudio==2.1.0
-torchmetrics
-transformers==4.34.1
-trimesh
-webdataset
-xatlas
-xformers
+accelerate==1.0.1
+bitsandbytes==0.44.1
+deepspeed
+diffusers==0.19.3
+einops
+huggingface-hub
+imageio[ffmpeg]
+invisible_watermark
+omegaconf
+peft
+plyfile
+PyMCubes
+pytorch-lightning==2.1.2
+rembg
+sentencepiece
+tensorboard
+torch==2.1.0 --index-url https://download.pytorch.org/whl/cu121
+torchvision==0.16.0
+torchaudio==2.1.0
+torchmetrics
+transformers==4.34.1
+trimesh
+webdataset
+xatlas
+xformers
 git+https://github.com/NVlabs/nvdiffrast/