accelerate==1.1.1
annotated-types==0.7.0
asttokens==2.4.1
certifi==2024.8.30
charset-normalizer==3.4.0
click==8.1.7
contourpy==1.3.1
cycler==0.12.1
decorator==5.1.1
diffusers==0.32.2
docker-pycreds==0.4.0
exceptiongroup==1.2.2
executing==2.1.0
filelock==3.13.1
fonttools==4.55.0
fsspec==2024.2.0
gitdb==4.0.11
gitpython==3.1.43
huggingface-hub==0.26.2
idna==3.10
importlib-metadata==8.5.0
ipdb==0.13.13
ipython==8.29.0
jedi==0.19.2
jinja2==3.1.3
joblib==1.4.2
kiwisolver==1.4.7
markupsafe==2.1.5
matplotlib==3.9.2
matplotlib-inline==0.1.7
mpmath==1.3.0
munch==4.0.0
networkx==3.2.1
numpy==1.26.3
packaging==24.2
pandas==2.2.3
parso==0.8.4
pexpect==4.9.0
pillow==10.2.0
platformdirs==4.3.6
prompt-toolkit==3.0.48
protobuf #==5.28.3
psutil==6.1.0
ptyprocess==0.7.0
pure-eval==0.2.3
pydantic==2.10.3
pydantic-core==2.27.1
pygments==2.18.0
pyparsing==3.2.0
python-dateutil==2.9.0.post0
pytz==2024.2
pyyaml==6.0.2
regex==2024.11.6
requests==2.32.3
safetensors==0.4.5
scikit-learn==1.6.0
scipy==1.14.1
seaborn==0.13.2
sentencepiece==0.2.0
sentry-sdk==2.19.2
setproctitle==1.3.4
six==1.16.0
smmap==5.0.1
stack-data==0.6.3
sympy==1.13.1
threadpoolctl==3.5.0
tokenizers==0.20.3
tomli==2.1.0
torch==2.4.1
torchaudio==2.4.1
torchvision==0.19.1
tqdm==4.67.1
traitlets==5.14.3
transformers==4.46.3
triton==3.0.0
typing-extensions==4.12.2
tzdata==2024.2
urllib3==2.2.3
wandb==0.19.0
wcwidth==0.2.13
zipp==3.21.0
bitsandbytes #==0.45.1
pycocotools
pytorch-fid
torch-fidelity
torchmetrics
image-reward
open-clip-torch
git+https://github.com/openai/CLIP.git
hpsv2
lpips