radames commited on
Commit
ad710cd
1 Parent(s): ff3a12f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -27,7 +27,7 @@ ENV HOME=/home/user \
27
  WORKDIR $HOME/app
28
 
29
  # download falcon-40b
30
- RUN python -c "from transformers import AutoModel; AutoModel.from_pretrained('tiiuae/falcon-40b', trust_remote_code=True)"
31
 
32
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
33
  COPY --chown=user . $HOME/app
 
27
  WORKDIR $HOME/app
28
 
29
  # download falcon-40b
30
+ RUN python -c "from transformers import AutoModel; AutoModel.from_pretrained('tiiuae/falcon-7b', trust_remote_code=True)"
31
 
32
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
33
  COPY --chown=user . $HOME/app