riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Hardian Lawi 2021-10-12 11:30:03 +08:00 committed by GitHub
parent 990de2c17c
commit 85d69a7dd1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/benchmark/benchmark_args_tf.py
View File

@ -79,6 +79,7 @@ class TensorFlowBenchmarkArguments(BenchmarkArguments):
@cached_property
@tf_required
def _setup_tpu(self) -> Tuple["tf.distribute.cluster_resolver.TPUClusterResolver"]:
tpu = None
if self.tpu:
try:
if self.tpu_name: