zeus-ml==0.4.0
fschat==0.2.20
rwkv==0.7.5
einops
tyro