Hot fix

AmpereComputingAI · Sep 25, 2023 · e4719bf · e4719bf
1 parent 3a6a461
commit e4719bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/natural_language_processing/extractive_question_answering/bert_large/run_huggingface.py b/natural_language_processing/extractive_question_answering/bert_large/run_huggingface.py
@@ -4,7 +4,6 @@
 import argparse
 
 import numpy as np
-import tensorflow as tf
 from transformers import AutoTokenizer, TFAutoModelForQuestionAnswering
 
 from utils.benchmark import run_model
@@ -38,6 +37,7 @@ def parse_args():
 
 
 def run_tf(model_name, batch_size, num_runs, timeout, squad_path):
+    import tensorflow as tf
     from utils.tf import TFSavedModelRunner
 
     def run_single_pass(tf_runner, squad):