requirements.txt

wheel
# torch==2.1.2
# torchvision==0.16.2
transformers==4.39.3
accelerate==0.30.0
diffusers>=0.30.1
numpy==1.24.4
einops
ftfy
ipython
contexttimer
opencv-python-headless==4.10.0.84
imageio==2.33.1
imageio-ffmpeg==0.5.1
packaging
pandas
plotly
pre-commit
python-magic
scikit-image
sentencepiece
spacy==3.7.5
streamlit
timm==0.6.12
tqdm
torchmetrics
tiktoken
jsonlines
tensorboardX
jupyter