SwissArmyTransformer>=0.3.6
torch>1.10.0
torchvision
transformers==4.33
mdtex2html
gradio
#bitsandbytes==0.39.0
wandb
SwissArmyTransformer==0.4.0