torch>=1.9.0
pytorch-lightning
transformers==4.28.1
datasets
evaluate
bitsandbytes==0.37.2
sentencepiece
deepspeed
gradio
click
wget
ai21
cohere
ipywidgets
openai>=0.27.0
pydantic>=1.10.0
rouge-score>=0.1.2
accelerate
wandb

[int4]
torch>=2.0
