minhtoan
/

gpt3-small-finetune-cnndaily-news

Text Generation

Inference Endpoints

Model card Files Files and versions Community

minhtoan commited on Feb 25, 2023

Commit

8a91ea4

•

1 Parent(s): 5e42dcd

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -29,8 +29,8 @@ Pretrained GPT-3 small, it's architecture intentionally resembles that of GPT-3,
 ~~~~
 from transformers import GPT2Tokenizer, GPTNeoForCausalLM
-tokenizer = GPT2Tokenizer.from_pretrained('gpt3-small-finetune-cnndaily-news-news')
-model = GPTNeoForCausalLM.from_pretrained('gpt3-small-finetune-cnndaily-news-news')
 text = "Ever noticed how plane seats appear to be getting smaller and smaller? "
 input_ids = tokenizer.encode(text, return_tensors='pt')

 ~~~~
 from transformers import GPT2Tokenizer, GPTNeoForCausalLM
+tokenizer = GPT2Tokenizer.from_pretrained('gpt3-small-finetune-cnndaily-news')
+model = GPTNeoForCausalLM.from_pretrained('gpt3-small-finetune-cnndaily-news')
 text = "Ever noticed how plane seats appear to be getting smaller and smaller? "
 input_ids = tokenizer.encode(text, return_tensors='pt')