Fixed bug in QAProcessor model name setting.
@@ -95,11 +95,11 @@ class QAProcessor:
def __init__(
self, model_name="bert-base-cased", to_lower=False, custom_tokenize=None, cache_dir="."
):
+ self.model_name = model_name
self.tokenizer = TOKENIZER_CLASS[model_name].from_pretrained(
model_name, do_lower_case=to_lower, cache_dir=cache_dir
)
self.custom_tokenize = custom_tokenize
- self.model_name = model_name
@property
def model_name(self):
