wangyoucao577 commited on
Commit
2152678
·
verified ·
1 Parent(s): 0a7574c

feat: use huggingface example

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -4
Dockerfile CHANGED
@@ -1,6 +1,6 @@
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
@@ -19,8 +19,9 @@ RUN . $NVM_DIR/nvm.sh && nvm install node
19
 
20
 
21
  # install TEN-Agent
22
- RUN git clone https://github.com/TEN-framework/TEN-Agent.git /app
23
- RUN cd /app && task use && cd agents && ./scripts/package.sh
 
24
  RUN . $NVM_DIR/nvm.sh && cd /app/playground && npm i
25
 
26
 
@@ -45,7 +46,6 @@ WORKDIR $HOME/app
45
  RUN pip install --no-cache-dir --upgrade pip
46
 
47
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
48
- # COPY --chown=user . $HOME/app
49
  COPY --chown=user .env /app/
50
  COPY --chown=user entrypoint.sh $HOME/app/
51
 
 
1
  FROM ghcr.io/ten-framework/ten_agent_build:0.5.0-cuda
 
2
 
3
+ # install huggingface cli
4
  RUN pip install "huggingface_hub[cli]" hf_transfer
5
 
6
  # install vllm
 
19
 
20
 
21
  # install TEN-Agent
22
+ # TODO: don't use branch
23
+ RUN git clone -b feature/hf https://github.com/TEN-framework/TEN-Agent.git /app
24
+ RUN cd /app && task use AGENT=agents/examples/huggingface && cd agents && ./scripts/package.sh
25
  RUN . $NVM_DIR/nvm.sh && cd /app/playground && npm i
26
 
27
 
 
46
  RUN pip install --no-cache-dir --upgrade pip
47
 
48
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 
49
  COPY --chown=user .env /app/
50
  COPY --chown=user entrypoint.sh $HOME/app/
51