# Core torch libraries (pinned first to prevent overrides)
torch==2.2.2
torchvision==0.17.2
triton==2.2.0

# Newly added libraries (versions known to be compatible with torch 2.3.1)
torchmetrics==1.6.2
pandas==2.2.2
scikit-learn==1.4.2
torch-fidelity==0.3.0
xformers==0.0.25.post1
datasets
webdataset
diffusers
peft

# Original dependencies
accelerate==1.1.1
antlr4-python3-runtime==4.9.3
certifi==2025.1.31
charset-normalizer==3.4.1
click==8.1.8
git+https://github.com/openai/CLIP.git
dictdot==1.5.1
docker-pycreds==0.4.0
einops==0.8.0
filelock==3.18.0
fsspec==2025.3.0
ftfy==6.3.1
gitdb==4.0.12
gitpython==3.1.44
h5py==3.12.1
huggingface-hub==0.29.3
idna==3.10
jinja2==3.1.6
markupsafe==3.0.2
mpmath==1.3.0
networkx==3.2.1
numpy==1.26.4
omegaconf==2.3.0
packaging==24.2
pillow==11.1.0
platformdirs==4.3.7
protobuf==5.29.4
psutil==7.0.0
pyyaml==6.0.2
regex==2024.11.6
requests==2.32.3
safetensors==0.5.3
sentry-sdk==2.24.0
setproctitle==1.3.5
six==1.17.0
smmap==5.0.2
sympy==1.13.3
timm==1.0.11
tokenizers==0.20.3
tqdm==4.67.1
transformers==4.46.3
typing-extensions==4.12.2
urllib3==2.3.0
wandb==0.18.7
wcwidth==0.2.13 
hydra-core 
flow_matching 
pycocotools
