transformers
AutoModelForCausalLM
AutoTokenizer
torch
AutoModel