lIlBrother commited on
Commit
b3086d6
1 Parent(s): 26dbdb8

Update: 오타수정

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -108,10 +108,10 @@ from transformers.pipelines import Text2TextGenerationPipeline
108
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
109
  texts = ["그러게 누가 6시까지 술을 마시래?"]
110
  tokenizer = AutoTokenizer.from_pretrained(
111
- inference_args.model_name_or_path,
112
  )
113
  model = AutoModelForSeq2SeqLM.from_pretrained(
114
- inference_args.model_name_or_path,
115
  )
116
  # BartText2TextGenerationPipeline is implemented above (see 'Use')
117
  seq2seqlm_pipeline = BartText2TextGenerationPipeline(model=model, tokenizer=tokenizer)
 
108
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
109
  texts = ["그러게 누가 6시까지 술을 마시래?"]
110
  tokenizer = AutoTokenizer.from_pretrained(
111
+ args.model_name_or_path,
112
  )
113
  model = AutoModelForSeq2SeqLM.from_pretrained(
114
+ args.model_name_or_path,
115
  )
116
  # BartText2TextGenerationPipeline is implemented above (see 'Use')
117
  seq2seqlm_pipeline = BartText2TextGenerationPipeline(model=model, tokenizer=tokenizer)