From bc5dfe00334d0db3ae12006b7d61d102f5f7f6aa Mon Sep 17 00:00:00 2001
From: GuanLuo <41310872+GuanLuo@users.noreply.github.com>
Date: Wed, 12 Jul 2023 16:41:43 -0700
Subject: [PATCH] Fix PyTorch implicit model mounting in
 gen_qa_model_repository (#6054)

---
 qa/common/gen_qa_model_repository | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/qa/common/gen_qa_model_repository b/qa/common/gen_qa_model_repository
index 34ed7f2e18..500683e418 100755
--- 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"