timm
einops
accelerate
transformers==4.51.3
diffusers
opencv-python-headless
scipy
wandb
matplotlib
Pillow
tqdm
omegaconf
python-dotenv
ninja
ipykernel
wheel
triton-windows; sys_platform == "win32"
# flash-attn==2.7.4.post1
peft

--extra-index-url https://download.pytorch.org/whl/cu124
torch==2.6.0
torchvision==0.21.0