diff --git a/examples/run_tf_ner.py b/examples/run_tf_ner.py index 6aa0f4bc320..ef970d83901 100644 --- a/examples/run_tf_ner.py +++ b/examples/run_tf_ner.py @@ -9,7 +9,6 @@ import re import numpy as np import tensorflow as tf from absl import app, flags, logging -from fastprogress import master_bar, progress_bar from seqeval import metrics from transformers import ( @@ -29,6 +28,12 @@ from transformers import ( from utils_ner import convert_examples_to_features, get_labels, read_examples_from_file +try: + from fastprogress import master_bar, progress_bar +except ImportError: + from fastprogress.fastprogress import master_bar, progress_bar + + ALL_MODELS = sum( (tuple(conf.pretrained_config_archive_map.keys()) for conf in (BertConfig, RobertaConfig, DistilBertConfig)), () )