Skip to content

Commit

Permalink
make flake8 happy
Browse files Browse the repository at this point in the history
  • Loading branch information
cwharris committed Mar 13, 2024
1 parent f081ceb commit 33ea700
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions examples/doca/run.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,12 +119,13 @@ def run_pipeline(pipeline_batch_size,
pipeline.add_stage(DeserializeStage(config))
pipeline.add_stage(MonitorStage(config, description="Deserialize rate", unit='pkts'))

hashfile = '/workspace/models/training-tuning-scripts/sid-models/resources/bert-base-uncased-hash.txt'

# add preprocessing stage
pipeline.add_stage(
PreprocessNLPStage(
config,
vocab_hash_file=
'/workspace/models/training-tuning-scripts/sid-models/resources/bert-base-uncased-hash.txt',
vocab_hash_file=hashfile,
do_lower_case=True,
truncation=True,
add_special_tokens=False,
Expand Down

0 comments on commit 33ea700

Please sign in to comment.