torch
transformers
tokenizers
accelerate