diff --git a/llm2clip/requirements.txt b/llm2clip/requirements.txt index ef28f8c..eda43c4 100644 --- a/llm2clip/requirements.txt +++ b/llm2clip/requirements.txt @@ -1,6 +1,6 @@ torch>=1.9.0 torchvision -deepspeed==0.6.5 +deepspeed==0.15.1 # xformers # apex # timm==0.5.4