gpu / requirements.txt
vitorcalvi's picture
1
54939b3
sentencepiece
tiktoken
jsonlines
diffusers==0.30.1
accelerate==0.30.0
torchvision
numpy==1.26.4
imageio
imageio-ffmpeg
timm
transformers
opencv-python-headless
einops
tensorboardX
ipython
# CUDA-enabled PyTorch packages
torchaudio
-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
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