Add max mem to 15 (#255)

marqo-ai · Sep 13, 2024 · cba2a33 · cba2a33
1 parent c34eab0
commit cba2a33
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/open-source-unit-tests.yml b/.github/workflows/open-source-unit-tests.yml
@@ -116,6 +116,8 @@ jobs:
           docker pull ${{ steps.prepare.outputs.registry }}/${{ steps.prepare.outputs.image_repo }}:${{ steps.prepare.outputs.image_tag }}
           docker run --name marqo -d --privileged -p 8882:8882 --add-host host.docker.internal:host-gateway \
             -e MARQO_ENABLE_BATCH_APIS=True \
+            -e MARQO_MAX_CUDA_MODEL_MEMORY=15 \
+            -e MARQO_MAX_CPU_MODEL_MEMORY=15 \
             ${{ steps.prepare.outputs.registry }}/${{ steps.prepare.outputs.image_repo }}:${{ steps.prepare.outputs.image_tag }}
                     
           # wait for marqo to start with timeout of 10 minutes