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)
|