non-torch-requirements-docker.txt

# do not run pip install -r requirements.txt directly under windows as cuda needs a specific version that's not available in the standard repository


# torch, torchvision, torchaudio need  
ipython==8.10.0
pandas==1.4.4
scikit-image==0.19.3
clean_fid==0.1.35
ftfy==6.1.1
scipy==1.10.0
transformers==4.26.1
omegaconf==2.3.0
einops==0.4.1
pytorch-lightning==1.7.7
tensorflow_hub
torchmetrics==0.11.1
kornia==0.6
accelerate==0.16.0
jsonmerge==1.9.0
resize-right==0.0.2
torchdiffeq==0.2.3
opencv-python==4.7.0.68
numpngw==0.1.2
triton; sys_platform == 'linux'
open-clip-torch==2.13.0
torchsde==0.2.5
scikit-learn==1.2.1
matplotlib==3.6.3
omegaconf==2.3.0
numexpr==2.8.4

# FILM dependencies. FILM uses Tensorflow
#tensorflow-gpu
tensorflow-datasets==4.4.0
tensorflow-hub
tensorflow-addons
gin-config==0.5.0
parameterized==0.8.1
mediapy==1.0.3
apache-beam

# other audio processing requirements
onnxruntime
git+https://github.com/linto-ai/whisper-timestamped#egg=whisper-timestamped
whisper
moviepy

# this is a fork of demucs that is stable while the upstream version's api settles down
git+https://github.com/boxabirds/demucs#egg=demucs
tdqm
librosa
openai
pydub
soundfile