Elastic-forcing-code / 1.3B /requirements.txt
Elastic-forcing's picture
Add training and inference source code
2ea2835 verified
Raw History Blame Contribute Delete
506 Bytes
# Training and inference dependencies. Install torch/torchvision first.
numpy==1.26.4
diffusers==0.31.0
transformers==4.53.3
tokenizers==0.21.4
timm==1.0.28
accelerate==1.14.0
tqdm==4.70.0
imageio==2.37.4
imageio-ffmpeg==0.6.0
easydict==1.13
ftfy==6.3.1
omegaconf>=2.3.0,<2.4
einops==0.8.1
av==13.1.0
opencv-python
open-clip-torch
lmdb==2.3.0
matplotlib
sentencepiece==0.2.2
scikit-image
huggingface-hub==0.36.2
safetensors==0.8.0
PyYAML==6.0.3
pandas
scipy
Pillow==12.2.0
requests==2.34.2
packaging==26.3