requirements.txt

accelerate==0.25.0
beautifulsoup4==4.12.2
bitsandbytes==0.41.3.post2
braceexpand==0.1.7
clip_retrieval==2.37.0
compel==2.0.2
d4rl==1.1
datasets==2.16.1
deepspeed==0.12.6
diffusion==6.10.1
einops==0.7.0
faiss==1.5.3
flax==0.7.5
ftfy==6.1.3
GitPython==3.1.40
gym==0.26.2
huggingface_hub==0.20.1
importlib_metadata==7.0.1
imwatermark==0.0.2
intel_extension_for_pytorch==2.1.100
jax==0.4.23
k_diffusion==0.1.1.post1
kornia==0.7.1
librosa==0.10.1
ligo==1.0.2
matplotlib==3.8.0
msgpack_python==0.5.6
muse==0.1
neural_compressor==2.4.1
note_seq==0.0.5
numpy==1.26.3
omegaconf==2.3.0
OmegaConf==2.3.0
onnx==1.15.0
onnx_graphsurgeon==0.0.1.dev5
optax==0.1.7
packaging==23.2
pandas==2.1.4
parameterized==0.9.0
peft==0.7.0
Pillow==10.0.1
Pillow==10.2.0
polygraphy==0.49.0
prodigyopt==1.0
pycuda==2024.1
PyGithub==2.1.1
pytest==7.4.4
PyYAML==6.0.1
PyYAML==6.0.1
Requests==2.31.0
requests_mock==1.11.0
safetensors==0.4.1
scipy==1.11.4
setuptools==68.2.2
t5x==0.0.0
tensorrt==8.6.1.post1
torch==2.1.2
torch_xla==2.1.0
torchsde==0.2.6
torchvision==0.16.2
tqdm==4.66.1
transformers==4.36.2
wandb==0.16.1
webdataset==0.2.86
xformers==0.0.23.post1