torch==2.1.0
numpy==1.24.3
transformers==4.36.0
einops==0.7.0
pyyaml==6.0.1
tokenizers>=0.15.0
gradio==4.44.0