Files changed (1) hide show
  1. README.md +2 -2
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')