This commit is contained in:
Martin Malmsten 2020-02-23 21:43:31 +01:00
parent 869b66f6b3
commit 33eb8a165d

View File

@ -617,7 +617,7 @@ def main():
tokenizer = tokenizer_class.from_pretrained(
args.tokenizer_name if args.tokenizer_name else args.model_name_or_path,
cache_dir=args.cache_dir if args.cache_dir else None,
**tokenizer_args
**tokenizer_args,
)
model = model_class.from_pretrained(
args.model_name_or_path,