"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "629b22adcfe340c4e3babac83654da2fbd1bbf89"
Commit a368b877 authored by Peng Qi's avatar Peng Qi
Browse files

Fix #1015

parent c589862b
...@@ -498,7 +498,7 @@ def main(): ...@@ -498,7 +498,7 @@ def main():
# Load a trained model and vocabulary that you have fine-tuned # Load a trained model and vocabulary that you have fine-tuned
model = model_class.from_pretrained(args.output_dir) model = model_class.from_pretrained(args.output_dir)
tokenizer = tokenizer_class.from_pretrained(args.output_dir) tokenizer = tokenizer_class.from_pretrained(args.output_dir, do_lower_case=args.do_lower_case)
model.to(args.device) model.to(args.device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment