transformers
tiktoken
flash-attn
torch
pillow