coderuday21 commited on
Commit
b7a71f3
·
1 Parent(s): 6777e8b

Fix: add einops dependency required by AdaptFormer custom code

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +1 -0
Dockerfile CHANGED
@@ -19,7 +19,7 @@ WORKDIR /app
19
 
20
  # Build-time info + cache-bust:
21
  # Changing APP_BUILD forces Docker to re-run subsequent layers (including pip install).
22
- ARG APP_BUILD=15
23
  ENV APP_BUILD=${APP_BUILD}
24
  RUN echo "Docker build start: APP_BUILD=${APP_BUILD}" && python -V
25
 
 
19
 
20
  # Build-time info + cache-bust:
21
  # Changing APP_BUILD forces Docker to re-run subsequent layers (including pip install).
22
+ ARG APP_BUILD=16
23
  ENV APP_BUILD=${APP_BUILD}
24
  RUN echo "Docker build start: APP_BUILD=${APP_BUILD}" && python -V
25
 
requirements.txt CHANGED
@@ -2,6 +2,7 @@
2
  torch
3
  torchvision
4
  transformers>=4.35.0
 
5
  fastapi>=0.104.0
6
  uvicorn[standard]>=0.24.0
7
  python-multipart>=0.0.6
 
2
  torch
3
  torchvision
4
  transformers>=4.35.0
5
+ einops
6
  fastapi>=0.104.0
7
  uvicorn[standard]>=0.24.0
8
  python-multipart>=0.0.6