diff --git a/session/mistral/requirements.txt b/session/mistral/requirements.txt index 54dad6a1..4a0b39e6 100644 --- a/session/mistral/requirements.txt +++ b/session/mistral/requirements.txt @@ -4,7 +4,7 @@ bitsandbytes==0.41.1 transformers==4.34.0 git+https://github.com/malaysia-ai/trl tokenizers==0.14.1 -deepspeed==0.10.0 +deepspeed==0.15.1 lightning==2.0.7 lightning-cloud==0.5.37 lightning-utilities==0.9.0