Update requirements.txt

requirements.txt

@@ -1,58 +1,36 @@
-# Core dependencies from pyproject.toml
-torch
-torchvision==0.18.0
+# Core dependencies
+uv
+hydra-core
+loguru
+Pillow
+pydub
+torch==2.3.0+cu128
+torchvision==0.18.0+cu128
+--extra-index-url https://download.pytorch.org/whl/cu128
+
+# FlashAttention2
+flash_attn==2.7.3
+
+# Transformers and training utilities
 transformers==4.46.0
-tokenizers>=0.15.2
-sentencepiece==0.1.99
-shortuuid
+pytorchvideo==0.1.5
+deepspeed==0.15.4
 accelerate==0.34.2
-peft>=0.9.0
-bitsandbytes==0.43.2
-pydantic>=1,<2
-markdown2[all]
 numpy==1.26.4
-scikit-learn==1.2.2
-gradio==3.35.2
-gradio_client==0.2.9
-requests
-httpx
-uvicorn
-fastapi
-fire
-seaborn
-ring_flash_attn==0.1.1
-einops==0.6.1
-einops-exts==0.0.4
-timm==0.9.12
-openpyxl==3.1.2
-pytorchvideo==0.1.5
-decord==0.6.0
-datasets==2.16.1
-openai==1.8.0
-webdataset==0.2.86
-nltk==3.3
-pywsd==1.2.4
 opencv-python-headless==4.8.0.76
-tyro
-pytest
-pre-commit
-loguru
-hydra-core
-xgrammar
-deepspeed==0.9.5
-wandb
-ninja
+matplotlib
 
-# Direct Git dependency
-git+https://github.com/bfshi/scaling_on_scales#egg=s2wrapper
-
-# Additional dependencies from shell script
+# Audio
 soundfile
 librosa
 openai-whisper
 ftfy
 ffmpeg
 jiwer
+wandb
 kaldiio
+peft==0.14.0
+
+# Compatibility fix
 protobuf==3.20.*
-triton==2.3.0  # Override to latest only if using FP8 quantization
+triton==3.1.0