kfkas
/

t5-large-korean-news-title-klue-ynat

Text2Text Generation

generated_from_keras_callback

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

kfkas commited on May 30, 2023

Commit

65c91f2

•

1 Parent(s): 8f6c674

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -20,7 +20,7 @@ OUTPUT = 각 label에 맞는 뉴스 기사 제목을 생성합니다.
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model_dir = "t5-base-korean-news-title-klue-ynat"
 tokenizer = AutoTokenizer.from_pretrained(model_dir)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_dir)
 model.to(device)

 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model_dir = "kfkas/t5-base-korean-news-title-klue-ynat"
 tokenizer = AutoTokenizer.from_pretrained(model_dir)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_dir)
 model.to(device)