1

requirements.txt

@@ -1,50 +1,15 @@
-# CUDA-enabled PyTorch packages
-torch==2.0.1+cu118
-torchvision==0.15.2+cu118
-torchaudio==2.0.2+cu118
--f https://download.pytorch.org/whl/torch_stable.html
-
-# Core dependencies
-gradio==4.38.1
-gradio_client==1.1.0
-
-# Additional dependencies
-absl-py==2.1.0
-aiofiles==23.2.1
-altair==5.3.0
-anyio==4.4.0
-attrs==23.2.0
-audioread==3.0.1
-certifi==2024.7.4
-charset-normalizer==3.3.2
-click==8.1.7
-decorator==4.4.2
-fastapi==0.111.1
-h5py==3.11.0
-huggingface-hub==0.23.5
-idna==3.7
-Jinja2==3.1.4
-joblib==1.4.2
-jsonschema==4.23.0
-kiwisolver==1.4.5
-librosa==0.10.2.post1
-MarkupSafe==2.1.5
-matplotlib==3.9.1
+sentencepiece
+tiktoken
+jsonlines
+diffusers==0.30.1
+accelerate==0.30.0
+torchvision
 numpy==1.26.4
-pandas==2.2.2
-Pillow==10.4.0
-pydantic==2.8.2
-python-multipart==0.0.9
-pytz==2024.1
-PyYAML==6.0.1
-requests==2.32.3
-scikit-learn==1.5.1
-scipy==1.14.0
-soundfile==0.12.1
-starlette==0.37.2
-tqdm==4.66.4
-transformers==4.42.4
-uvicorn==0.30.1
-
-# Any other necessary dependencies
-# Add your additional dependencies here
+imageio
+imageio-ffmpeg
+timm
+transformers
+opencv-python-headless
+einops
+tensorboardX
+ipython