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