Skip to content

Commit ce7a7a6

Browse files
GuanLuomc-nv
authored andcommitted
Fix PyTorch implicit model mounting in gen_qa_model_repository (#6054)
1 parent 9889f0c commit ce7a7a6

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

qa/common/gen_qa_model_repository

+2
Original file line numberDiff line numberDiff line change
@@ -371,6 +371,8 @@ docker run $DOCKER_GPU_ARGS --rm --entrypoint $SRCDIR/$TORCHSCRIPT \
371371
--mount type=bind,source=$HOST_VARSEQDESTDIR,target=$VARSEQDESTDIR \
372372
--mount type=bind,source=$HOST_TORCHTRTDESTDIR,target=$TORCHTRTDESTDIR \
373373
--mount type=bind,source=$HOST_RAGGEDDESTDIR,target=$RAGGEDDESTDIR \
374+
--mount type=bind,source=$HOST_VARIMPLICITSEQDESTDIR,target=$VARIMPLICITSEQDESTDIR \
375+
--mount type=bind,source=$HOST_IMPLICITSEQDESTDIR,target=$IMPLICITSEQDESTDIR \
374376
$PYTORCH_IMAGE
375377
if [ $? -ne 0 ]; then
376378
echo -e "Failed"

0 commit comments

Comments
 (0)