requirements.txt

pydantic==2.10.6
torch==2.6.0
einops==0.8.1
transformers==4.49.0
huggingface-hub==0.34.4
omegaconf==2.3.0
sentencepiece==0.2.0
opencv-python
matplotlib
onnxruntime
torchvision==0.21.0
invisible-watermark==0.2.0
fire
tqdm
numpy==2.2.4