torch
numpy
transformers
wandb
tqdm
streamlit
tiktoken==0.4.0