Use MorpheusRelativePath for locating the model_vocab_hash_file

nv-morpheus · Jan 10, 2024 · eb36baa · eb36baa
1 parent 0a297c6
commit eb36baa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/examples/log_parsing/README.md b/examples/log_parsing/README.md
@@ -63,7 +63,7 @@ python run.py \
     --num_threads 1 \
     --input_file ${MORPHEUS_ROOT}/models/datasets/validation-data/log-parsing-validation-data-input.csv \
     --output_file ./log-parsing-output.jsonlines \
-    --model_vocab_hash_file=${MORPHEUS_ROOT}/morpheus/data/bert-base-cased-hash.txt \
+    --model_vocab_hash_file=data/bert-base-cased-hash.txt \
     --model_vocab_file=${MORPHEUS_ROOT}/models/training-tuning-scripts/sid-models/resources/bert-base-cased-vocab.txt \
     --model_seq_length=256 \
     --model_name log-parsing-onnx \

diff --git a/examples/log_parsing/run.py b/examples/log_parsing/run.py
@@ -18,6 +18,7 @@
 from inference import LogParsingInferenceStage
 from postprocessing import LogParsingPostProcessingStage
 
+from morpheus.cli.utils import MorpheusRelativePath
 from morpheus.config import Config
 from morpheus.config import PipelineModes
 from morpheus.pipeline import LinearPipeline
@@ -61,7 +62,7 @@
 )
 @click.option('--model_vocab_hash_file',
               required=True,
-              type=click.Path(exists=True, dir_okay=False),
+              type=MorpheusRelativePath(exists=True, dir_okay=False),
               help="Model vocab hash file to use for pre-processing.")
 @click.option('--model_vocab_file',
               required=True,