Skip to content

Commit a8f122d

Browse files
authored
Fix PyTorch implicit model mounting in gen_qa_model_repository (#6054)
1 parent ee8d048 commit a8f122d

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)