Hugging Face's logo

Spaces:
maomao88
/
model_structure_viewer
Running

App Files Community
maomao88 commited on
Commit
e572672
·
1 Parent(s): 0c7c014

install einops flash_attn after requirements

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -18,6 +18,7 @@ WORKDIR /app
18
  # Copy backend requirements
19
  COPY --chown=user backend/requirements.txt /app/requirements.txt
20
  RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt
 
21
 
22
  # Copy backend and frontend
23
  COPY --chown=user backend/ /app/backend/
 
18
  # Copy backend requirements
19
  COPY --chown=user backend/requirements.txt /app/requirements.txt
20
  RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt
21
+ RUN pip install --no-cache-dir einops flash_attn
22
 
23
  # Copy backend and frontend
24
  COPY --chown=user backend/ /app/backend/