Commit
•
b3086d6
1
Parent(s):
26dbdb8
Update: 오타수정
Browse files
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 |
-
|
112 |
)
|
113 |
model = AutoModelForSeq2SeqLM.from_pretrained(
|
114 |
-
|
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)
|