requirements.txt

torch
torchvision
torchdiffeq>=0.2.3
torchsde>=0.2.6
einops>=0.6.0
open-clip-torch>=2.24.0
transformers>=4.29.1
tokenizers>=0.13.3
sentencepiece
peft>=0.10.0
torchinfo
safetensors>=0.4.2
bitsandbytes>=0.43.0 ;platform_system != 'Darwin'
bitsandbytes ;platform_system == 'Darwin'
aiohttp>=3.8.4
accelerate>=0.25.0
pyyaml>=6.0
scikit-image>=0.20.0
jsonmerge>=1.9.0
clean-fid>=0.1.35
resize-right>=0.0.2
opencv-python-headless>=4.9.0.80
albumentations>=1.3.0
aiofiles>=23.1.0
frozendict>=2.3.6
python-dateutil>=2.8.2
importlib_resources
Pillow
scipy
tqdm
protobuf>=3.20.0,<5.0.0
psutil
ConfigArgParse
aio-pika
pyjwt[crypto]
kornia>=0.7.0
mpmath>=1.0,!=1.4.0a0
huggingface_hub
lazy-object-proxy
lazy_loader>=0.3
can_ada
fsspec
natsort
OpenEXR
opentelemetry-distro
opentelemetry-exporter-otlp<=1.27.0
opentelemetry-propagator-jaeger
opentelemetry-instrumentation
opentelemetry-util-http
opentelemetry-instrumentation-aio-pika
opentelemetry-instrumentation-requests
opentelemetry-semantic-conventions
wrapt>=1.16.0
certifi
spandrel>=0.3.4
numpy>=1.24.4,<2.0.0
soundfile
watchdog
PySoundFile
networkx>=2.6.3
joblib
jaxtyping
spandrel_extra_arches
ml_dtypes
diffusers>=0.30.1
vtracer
skia-python
pebble>=5.0.7
openai
anthropic
humanize