Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained(
4 "oz1115/korean-gpt-quick-test",
5 trust_remote_code=True
6)
7
8model = AutoModelForCausalLM.from_pretrained(
9 "oz1115/korean-gpt-quick-test",
10 trust_remote_code=True
11)
12
13inputs = tokenizer("안녕하세요", return_tensors="pt")
14outputs = model.generate(**inputs, max_length=50)
15print(tokenizer.decode(outputs[0]))