alessandro trinca tornidor commited on
Commit
ac6ac74
1 Parent(s): 0914710

[bug] handle SAM machine learning models folder

Browse files
Files changed (2) hide show
  1. Dockerfile +3 -1
  2. dockerfiles/dockerfile-lisa-base +3 -1
Dockerfile CHANGED
@@ -61,6 +61,7 @@ RUN ls -ld ${LAMBDA_TASK_ROOT}/
61
  RUN poetry run python -m pip install pip wheel setuptools --upgrade
62
  RUN poetry install --with ${DEPENDENCY_GROUP} --no-root -vvv
63
 
 
64
 
65
  FROM nvcr.io/nvidia/pytorch:24.01-py3 as runtime
66
 
@@ -128,7 +129,8 @@ FROM runtime
128
  ARG FASTAPI_STATIC
129
  RUN mkdir ${FASTAPI_STATIC}
130
 
131
- COPY ./machine_learning_models ${LAMBDA_TASK_ROOT}/machine_learning_models
 
132
  COPY --from=node_prod_deps /appnode/node_modules* ${FASTAPI_STATIC}/node_modules
133
  COPY --from=node_build /appnode/dist* ${FASTAPI_STATIC}/dist
134
 
 
61
  RUN poetry run python -m pip install pip wheel setuptools --upgrade
62
  RUN poetry install --with ${DEPENDENCY_GROUP} --no-root -vvv
63
 
64
+ RUN git clone https://huggingface.co/aletrn/sam-quantized/ ${LAMBDA_TASK_ROOT}/sam-quantized
65
 
66
  FROM nvcr.io/nvidia/pytorch:24.01-py3 as runtime
67
 
 
129
  ARG FASTAPI_STATIC
130
  RUN mkdir ${FASTAPI_STATIC}
131
 
132
+ COPY --from=builder_global ${LAMBDA_TASK_ROOT}/sam-quantized/machine_learning_models \
133
+ ${LAMBDA_TASK_ROOT}/machine_learning_models
134
  COPY --from=node_prod_deps /appnode/node_modules* ${FASTAPI_STATIC}/node_modules
135
  COPY --from=node_build /appnode/dist* ${FASTAPI_STATIC}/dist
136
 
dockerfiles/dockerfile-lisa-base CHANGED
@@ -65,6 +65,7 @@ RUN ls -ld ${LAMBDA_TASK_ROOT}/
65
  RUN poetry run python -m pip install pip wheel setuptools --upgrade
66
  RUN poetry install --with ${DEPENDENCY_GROUP} --no-root -vvv
67
 
 
68
 
69
  FROM nvcr.io/nvidia/pytorch:24.01-py3 as runtime
70
 
@@ -135,6 +136,7 @@ FROM runtime
135
  ARG FASTAPI_STATIC
136
  RUN mkdir ${FASTAPI_STATIC}
137
 
138
- COPY ./machine_learning_models ${LAMBDA_TASK_ROOT}/machine_learning_models
 
139
  COPY --from=node_prod_deps /appnode/node_modules* ${FASTAPI_STATIC}/node_modules
140
  COPY --from=node_build /appnode/dist* ${FASTAPI_STATIC}/dist
 
65
  RUN poetry run python -m pip install pip wheel setuptools --upgrade
66
  RUN poetry install --with ${DEPENDENCY_GROUP} --no-root -vvv
67
 
68
+ RUN git clone https://huggingface.co/aletrn/sam-quantized/ ${LAMBDA_TASK_ROOT}/sam-quantized
69
 
70
  FROM nvcr.io/nvidia/pytorch:24.01-py3 as runtime
71
 
 
136
  ARG FASTAPI_STATIC
137
  RUN mkdir ${FASTAPI_STATIC}
138
 
139
+ COPY --from=builder_global ${LAMBDA_TASK_ROOT}/sam-quantized/machine_learning_models \
140
+ ${LAMBDA_TASK_ROOT}/machine_learning_models
141
  COPY --from=node_prod_deps /appnode/node_modules* ${FASTAPI_STATIC}/node_modules
142
  COPY --from=node_build /appnode/dist* ${FASTAPI_STATIC}/dist