diff --git a/requirements.txt b/requirements.txt index f3692468e..a8c01b53f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -17,7 +17,7 @@ sentencepiece wandb einops xformers>=0.0.22 -optimum +optimum==1.13.2 hf_transfer colorama numba