torch
transformers
bitsandbytes