Update requirements.txt

requirements.txt

@@ -1,3 +1,6 @@
+--extra-index-url https://download.pytorch.org/whl/cu121
+
+# --- Dependencias Base de Step1X-3D ---
 datasets==2.19.0
 diffusers==0.32.2
 einops==0.8.0
@@ -28,10 +31,6 @@ tqdm==4.66.2
 transformers==4.48.0
 trimesh==4.3.2
 spaces==0.28.3
-# --- CORRECCIONES ---
-accelerate>=0.28.0
-bitsandbytes>=0.41.0
-# --------------------
 rembg==2.0.65
 gradio==5.5.0
 wandb==0.18.6
@@ -53,4 +52,10 @@ kornia==0.8.0
 git+https://github.com/NVlabs/nvdiffrast.git
 torch==2.5.1
 torchvision==0.20.1
-torchaudio==2.5.1
+torchaudio==2.5.1
+
+accelerate>=0.28.0
+bitsandbytes>=0.41.0
+gguf
+xformers>=0.0.26
+https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl