josu
/

gpt-neo-1.3B-instruction

Text Generation

Inference Endpoints

Model card Files Files and versions Community

josu commited on Apr 26, 2023

Commit

9730f28

•

1 Parent(s): ad33c65

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -13,8 +13,8 @@ widget:
 from transformers import GenerationConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model = AutoModelForCausalLM.from_pretrained("josu/gpt-neo-br-instruction")
-tokenizer = AutoTokenizer.from_pretrained("josu/gpt-neo-br-instruction")
 def generate_prompt(instruction, input=None):

 from transformers import GenerationConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model = AutoModelForCausalLM.from_pretrained("josu/gpt-neo-1.3B-instruction")
+tokenizer = AutoTokenizer.from_pretrained("josu/gpt-neo-1.3B-instruction")
 def generate_prompt(instruction, input=None):