update setup

This commit is contained in:
cfli 2024-10-30 19:34:29 +08:00
parent 2d8345873e
commit bd868335e3

View File

@ -22,6 +22,7 @@ setup(
'peft',
'deepspeed',
'flash-attn==2.5.6',
'ir-datasets'
'ir-datasets',
'sentencepiece'
],
)