Fix PyTorch implicit model mounting in gen_qa_model_repository (#6054)

triton-inference-server · Jul 13, 2023 · 8e5c401 · 8e5c401
1 parent 966c0e2
commit 8e5c401
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/qa/common/gen_qa_model_repository b/qa/common/gen_qa_model_repository
@@ -371,6 +371,8 @@ docker run $DOCKER_GPU_ARGS --rm --entrypoint $SRCDIR/$TORCHSCRIPT \
        --mount type=bind,source=$HOST_VARSEQDESTDIR,target=$VARSEQDESTDIR \
        --mount type=bind,source=$HOST_TORCHTRTDESTDIR,target=$TORCHTRTDESTDIR \
        --mount type=bind,source=$HOST_RAGGEDDESTDIR,target=$RAGGEDDESTDIR \
+       --mount type=bind,source=$HOST_VARIMPLICITSEQDESTDIR,target=$VARIMPLICITSEQDESTDIR \
+       --mount type=bind,source=$HOST_IMPLICITSEQDESTDIR,target=$IMPLICITSEQDESTDIR \
        $PYTORCH_IMAGE
 if [ $? -ne 0 ]; then
     echo -e "Failed"