transformers==4.45.2
pytorch-lightning==2.2.1
accelerate==0.34.2
deepspeed==0.15.2
triton==2.3.1
diffusers==0.30.3
controlnet_aux==0.0.9
opencv-python==4.5.5.64
opencv-python-headless==4.5.5.64
pillow==10.2.0
mediapipe
flash-attn==2.6.3
flash-attention-fused-dense @ git+https://github.com/Dao-AILab/flash-attention@v2.6.3#subdirectory=csrc/fused_dense_lib
datasets==3.0.1
huggingface_hub==0.24
sentencepiece==0.1.99
timm==0.6.7
torchmetrics==1.3.2
einops
opt_einsum
pykeops
zstandard
h5py
wandb==0.16.3
hydra-core==1.3.1
hydra-colorlog==1.2.0
hydra-optuna-sweeper==1.2.0
pyrootutils
matplotlib
jupyter
ipython
rich
pytest
ipdb
gpustat
scikit-learn
spacy
munch
fvcore
psutil
twine
gdown
jaxtyping
gradio