accelerate
torch
transformers[AutoModelForCausalLM,AutoTokenizer]