abedmaatalla
commited on
Commit
•
4b3387e
1
Parent(s):
8f0cf73
Update README.md
Browse filesCorrecting the model name.
README.md
CHANGED
@@ -28,8 +28,8 @@ Pretrained GPT-3 small, continuing the development of GPT NEO, with architecture
|
|
28 |
~~~~
|
29 |
from transformers import GPT2Tokenizer, GPTNeoForCausalLM
|
30 |
|
31 |
-
tokenizer = GPT2Tokenizer.from_pretrained('gpt3-small-finetune-cnndaily-news')
|
32 |
-
model = GPTNeoForCausalLM.from_pretrained('gpt3-small-finetune-cnndaily-news')
|
33 |
|
34 |
text = "Ever noticed how plane seats appear to be getting smaller and smaller? "
|
35 |
input_ids = tokenizer.encode(text, return_tensors='pt')
|
|
|
28 |
~~~~
|
29 |
from transformers import GPT2Tokenizer, GPTNeoForCausalLM
|
30 |
|
31 |
+
tokenizer = GPT2Tokenizer.from_pretrained('minhtoan/gpt3-small-finetune-cnndaily-news')
|
32 |
+
model = GPTNeoForCausalLM.from_pretrained('minhtoan/gpt3-small-finetune-cnndaily-news')
|
33 |
|
34 |
text = "Ever noticed how plane seats appear to be getting smaller and smaller? "
|
35 |
input_ids = tokenizer.encode(text, return_tensors='pt')
|