wangyoucao577 commited on
Commit
ba8572c
·
verified ·
1 Parent(s): 95e409e

fix: use gpu

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -1,10 +1,10 @@
1
- #FROM ghcr.io/ten-framework/ten_agent_build:0.5.0-cuda
2
- FROM ghcr.io/ten-framework/ten_agent_build:0.5.0-2-g7d064cd-vllm-cpu
3
 
4
  RUN pip install "huggingface_hub[cli]" hf_transfer
5
 
6
  # install vllm
7
- #RUN pip install vllm
8
 
9
 
10
  # install node and npm for demo
 
1
+ FROM ghcr.io/ten-framework/ten_agent_build:0.5.0-cuda
2
+ #FROM ghcr.io/ten-framework/ten_agent_build:0.5.0-2-g7d064cd-vllm-cpu
3
 
4
  RUN pip install "huggingface_hub[cli]" hf_transfer
5
 
6
  # install vllm
7
+ RUN pip install vllm
8
 
9
 
10
  # install node and npm for demo