einops
flash-attn
torch
tqdm
transformers==4.34.0
