ejschwartz commited on
Commit
569beb5
1 Parent(s): 5b3f24d

Fix cache for downloading model from HF

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -35,7 +35,8 @@ RUN --mount=type=cache,target=/root/.cache pip install --break-system-packages -
35
  # Download the DIRTY model
36
  #RUN --mount=type=cache,target=/root/.cache (cd /root/.cache; (echo "b1e812b758eccf402271607c40fa491b5486742abf3706be174dc3f4fe87b9dd data1.tar.bz2" | sha256sum -c || wget -O '/root/.cache/data1.tar.bz2' 'https://cmu.box.com/shared/static/nx9fyn8jx0i9p4bftw8f2giqlufnoyj5')) && tar -xvjf /root/.cache/data1.tar.bz2 -C /DIRTY/dirty
37
  RUN pip install --break-system-packages huggingface_hub[cli]
38
- RUN --mount=type=cache,target=/root/.cache huggingface-cli download --repo-type model ejschwartz/dirty-ghidra --local-dir /DIRTY/dirty # 10-01-24
 
39
 
40
  # Set up a new user named "user"
41
  RUN useradd -m -o -u 1000 user
 
35
  # Download the DIRTY model
36
  #RUN --mount=type=cache,target=/root/.cache (cd /root/.cache; (echo "b1e812b758eccf402271607c40fa491b5486742abf3706be174dc3f4fe87b9dd data1.tar.bz2" | sha256sum -c || wget -O '/root/.cache/data1.tar.bz2' 'https://cmu.box.com/shared/static/nx9fyn8jx0i9p4bftw8f2giqlufnoyj5')) && tar -xvjf /root/.cache/data1.tar.bz2 -C /DIRTY/dirty
37
  RUN pip install --break-system-packages huggingface_hub[cli]
38
+
39
+ RUN --mount=type=cache,target=/tmp/model-dl huggingface-cli download --repo-type model ejschwartz/dirty-ghidra --local-dir /tmp/model-dl && cp -R /tmp/model-dl/data1 /DIRTY/dirty/data1
40
 
41
  # Set up a new user named "user"
42
  RUN useradd -m -o -u 1000 user