Hugging Face's logo

Spaces:
radames
/
Falcon-40b-Dockerfile
Build error

App Files Community
radames commited on
Commit
7ea9fd1
•
1 Parent(s): 7834358

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -27,8 +27,8 @@ ENV HOME=/home/user \
27
  WORKDIR $HOME/app
28
 
29
  # download falcon-40b
30
- RUN python -c 'from transformers import AutoModelForCausalLM; AutoModelForCausalLM.from_pretrained("tiiuae/falcon-40b", trust_remote_code=True, load_in_4bit=True, device_map="auto")'
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
34
 
 
27
  WORKDIR $HOME/app
28
 
29
  # download falcon-40b
30
+ RUN python -c 'from huggingface_hub import snapshot_download; snapshot_download("tiiuae/falcon-40b")'
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
34