[REQS]

The-Swarm-Corporation · Apr 17, 2024 · 8636adc · 8636adc
1 parent 92bf875
commit 8636adc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -19,8 +19,9 @@ peft
 accelerate
 transformers==4.37.2
 huggingface-hub
-optimum
+optimumgit ad
 auto-gptq
 whisperx
 shortuuid
-exxa
+exxa
+hf_transfer
diff --git a/servers/cogvlm/Dockerfile b/servers/cogvlm/Dockerfile
@@ -27,6 +27,7 @@ RUN python3.10 -m pip install -r requirements.txt
 RUN python3.10 -m pip install uvicorn
 RUN python3.10 -m pip install python-dotenv
 RUN python3.10 -m pip install torch
+RUN python3.10 -m pip install hf_transfer
 
 # Copy the application's entire directory structure into the container
 COPY . /swarms-cloud