matdmiller commited on
Commit
3b0c289
·
verified ·
1 Parent(s): 99fd5dc

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -29,7 +29,7 @@ ENTRYPOINT ["/bin/sh", "-c"]
29
  # curl test -H 'Authorization: Bearer $(cat /run/secrets/HF_TOKEN)'
30
 
31
  RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=true \
32
- python3 -c "from huggingface_hub import snapshot_download; import os; os.environ['HF_TOKEN']=$(cat /run/secrets/HF_TOKEN); snapshot_download(repo_id='LocalResearchGroup/smollm2-135m-20250224_023514', local_dir='.', allow_patterns='.aim/**', token=True)"
33
 
34
  # have to run `aim init` in the directory that stores aim data for
35
  # otherwise `aim up` will prompt for confirmation to create the directory itself.
 
29
  # curl test -H 'Authorization: Bearer $(cat /run/secrets/HF_TOKEN)'
30
 
31
  RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=true \
32
+ python3 -c "from huggingface_hub import snapshot_download; import os; with open('/run/secrets/HF_TOKEN', 'r') as f: token = f.read().strip(); snapshot_download(repo_id='LocalResearchGroup/smollm2-135m-20250224_023514', local_dir='.', allow_patterns='.aim/**', token=token)"
33
 
34
  # have to run `aim init` in the directory that stores aim data for
35
  # otherwise `aim up` will prompt for confirmation to create the directory itself.