heegyu
/

kodialogpt-v1

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

heegyu commited on Nov 7, 2022

Commit

5f817e2

•

1 Parent(s): 29cbfbb

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -11,8 +11,8 @@ license: cc-by-nc-sa-4.0
 ## 사용예시
 ```
-tokenizer = AutoTokenizer.from_pretrained("heegyu/kodialogpt")
-model = AutoModelForCausalLM.from_pretrained("heegyu/kodialogpt")
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 generation_args = dict(
     num_beams=4,

 ## 사용예시
 ```
+tokenizer = AutoTokenizer.from_pretrained("heegyu/kodialogpt-v1")
+model = AutoModelForCausalLM.from_pretrained("heegyu/kodialogpt-v1")
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 generation_args = dict(
     num_beams=4,