torch>=1.9.0
pytorch-lightning
transformers==4.31.0
datasets==2.14.5
evaluate==0.4.0
bitsandbytes==0.41.1
sentencepiece
deepspeed==0.9.5
gradio
click
wget
ai21
cohere
ipywidgets
openai>=0.27.0
pydantic>=1.10.0
rouge-score>=0.1.2
accelerate==0.22.0
wandb

[int4]
torch>=2.0
