Fix missing tpu variable in benchmark_args_tf.py (huggingface#13968)

Albertobegue · Jan 27, 2022 · 4613584 · 4613584
1 parent d7f5300
commit 4613584
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/benchmark/benchmark_args_tf.py b/src/transformers/benchmark/benchmark_args_tf.py
@@ -79,6 +79,7 @@ def __init__(self, **kwargs):
     @cached_property
     @tf_required
     def _setup_tpu(self) -> Tuple["tf.distribute.cluster_resolver.TPUClusterResolver"]:
+        tpu = None
         if self.tpu:
             try:
                 if self.tpu_name: