Update requirements.txt

requirements.txt

@@ -1,12 +1,14 @@
-datasets>=2.10.0
+
 trl>=0.4.7
 flash_attn==2.7.4.post1
-torch==2.6.0
 transformers==4.48.2
 accelerate==1.3.0
-soundfile==0.13.1
-pillow==11.1.0
-scipy==1.15.2
-torchvision==0.21.0
-backoff==2.2.1
 peft==0.13.2
+torch==1.13.1+cu117  # CUDA version might vary depending on your system
+datasets==2.10.1
+trfl==0.5.6
+tqdm==4.66.0
+tokenizers==0.13.2
+
+
+