einops
flash-attn
torch
tqdm
transformers==4.35.2
