Update requirements.txt

requirements.txt

@@ -1,35 +1,31 @@
+# Core ML Stack
 git+https://github.com/huggingface/diffusers.git
 transformers
 accelerate
 safetensors
-sentencepiece
 peft
-ftfy
-imageio-ffmpeg
+open_clip_torch
+
+# Data & Processing
+numpy
+Pillow
+scipy
+tqdm
 opencv-python
+imageio-ffmpeg
+av
+librosa
+soundfile
 
-torch == 2.4.0
-torchvision
-torchaudio
+# Utilities & Networking
 python-dotenv
-cython
-gitpython >= 3.1
-tensorboard >= 2.11
-numpy >= 1.21, <2.1
-Pillow >= 9.5
-scipy >= 1.7
-tqdm >= 4.66.1
-gradio >= 3.34
-einops >= 0.6
-hydra-core >= 1.3.2
+gitpython
 requests
+einops
+
+# Optional/Frameworks
+# Only include these if your code explicitly imports them
+hydra-core
 torchdiffeq
-librosa >= 0.8.1
-nitrous-ema
 auraloss
-hydra_colorlog
-tensordict
-colorlog
-open_clip_torch
-soundfile
-av
+sentencepiece