numpy
ipykernel
ipywidgets
torch
sentencepiece
transformers
accelerate
bitsandbytes