Update requirements.txt

requirements.txt

@@ -1,15 +1,47 @@
-huggingface_hub==0.25.2
-torch
-transformers
-gradio
-spaces
-opencv-python
-numpy
-einops
-Pillow
-sentencepiece
-peft
-bitsandbytes
-scipy
-tqdm
-accelerate
+# PyTorch and related libraries
+torch==2.2.0
+torchvision==0.17.0
+deepspeed==0.13.1
+accelerate==0.26.1
+peft==0.4.0
+timm==1.0.3
+
+# Hugging Face & NLP
+transformers==4.40.0
+tokenizers==0.19.1
+huggingface_hub==0.23.4
+sentencepiece==0.1.99
+
+# Data Processing and I/O
+numpy==1.24.4
+decord==0.6.0
+imageio==2.34.0
+imageio-ffmpeg==0.4.9
+moviepy==1.0.3
+opencv-python==4.6.0.66
+pysubs2
+
+# Miscellaneous
+scikit-learn==1.2.2
+shortuuid
+einops==0.6.1
+einops-exts==0.0.4
+bitsandbytes==0.43.0
+pydantic>=2.0
+markdown2[all]
+gradio==3.50.0
+gradio_client==0.6.1
+httpx==0.24.1
+requests
+openai
+uvicorn
+fastapi
+tensorboard
+wandb
+tabulate
+
+# Space Dependencies
+spaces==0.29.2
+
+# Optional CUDA for GPU Support
+--extra-index-url https://download.pytorch.org/whl/cu118