diff --git a/.dockerignore b/.dockerignore index 3cab3d6..ce6b302 100644 --- a/.dockerignore +++ b/.dockerignore @@ -1,7 +1,3 @@ -data -deployment -deployment-next - **/.git **/node_modules **/__pycache__ diff --git a/Dockerfile b/Dockerfile index 460b4f8..dcede3d 100644 --- a/Dockerfile +++ b/Dockerfile @@ -28,4 +28,4 @@ RUN poetry export --without-hashes -o requirements.txt RUN --mount=type=cache,target=/root/.cache pip install -i $PYPI_INDEX_URL --extra-index-url https://pypi.org/simple --no-dependencies -r requirements.txt -COPY . . +COPY tabby ./tabby diff --git a/deployment/scripts/triton.sh b/deployment/scripts/triton.sh index c86f3eb..bb0b05c 100755 --- a/deployment/scripts/triton.sh +++ b/deployment/scripts/triton.sh @@ -1,6 +1,9 @@ #!/bin/bash set -e +if [ -d "$MODEL_NAME" ]; then +MODEL_DIR="$MODEL_NAME" +else # Get model dir. MODEL_DIR=$(python3 <