requirements.txt

# Core dependencies
torch
torchvision
numpy
pillow
opencv-python
transformers
tokenizers
scipy
scikit-image
pyyaml
pandas
tqdm
webdataset
accelerate
xformers
bitsandbytes
jupyter
matplotlib
timm
flash_attn