Downloads · 30 days
13
4% of all-time downloads
kimdwan/polyglot-ko-1.3b-Logan
polyglot-ko-1.3b-Logan is a text generation model from kimdwan. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as unknown.
안녕하세요 문장생성 모델입니다. 데이터는 ai hub에 일반 전문분야 데이터로 학습하였습니다. ^^
Downloads · 30 days
13
4% of all-time downloads
All-time downloads
335
Public
Repo size
10.7 GB
Likes
2
Public
Click a slice to open those files.
.bin5.3 GB · 100%
From the Hugging Face model README
안녕하세요 문장생성 모델입니다. 데이터는 ai hub에 일반 전문분야 데이터로 학습하였습니다. ^^
!pip install transformers
!pip install sentence_transformers
from transformers import AutoTokenizer,GPTNeoXForCausalLM
path = "kimdwan/polyglot-ko-1.3b-Logan"
model = GPTNeoXForCausalLM.from_pretrained(path)
tokenizer = AutoTokenizer.from_pretrained(path)
# 넣고 싶은 단어등을 넣어주면 자동으로 문장을 생성해주나 20토큰 안에서 가능합니다.
text = "인공"
token = tokenizer(text,return_tensors='pt')
model.eval()
model = model.to("cpu")
output = model.generate(input_ids = token["input_ids"])
tokenizer.decode(output[0])