diff --git a/requirements.txt b/requirements.txt index dce85f9..4004b9b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -31,3 +31,6 @@ accelerate==0.30.1 socksio==1.0.0 gradio gradio_client +deepspeed==0.14.4 +peft==0.11.1 +tensorboard==2.17.0