flax-community
/

gpt-neo-125M-code-clippy-dedup

Text Generation

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

ncoop57 commited on Jul 23, 2021

Commit

71bd15f

•

1 Parent(s): b1b1363

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -23,9 +23,9 @@ You can use this model directly with a pipeline for text generation. This exampl
 from transformers import AutoModelForCausalLM, AutoTokenizer, FlaxAutoModelForCausalLM
-model = AutoModelForCausalLM.from_pretrained("flax-community/gpt-neo-125M-code-clippy-code-search-py")
-tokenizer = AutoTokenizer.from_pretrained("flax-community/gpt-neo-125M-code-clippy-code-search-py")
 prompt = """def greet(name):
   '''A function to greet user. Given a user name it should say hello'''

 from transformers import AutoModelForCausalLM, AutoTokenizer, FlaxAutoModelForCausalLM
+model = AutoModelForCausalLM.from_pretrained("flax-community/gpt-neo-125M-code-clippy-dedup")
+tokenizer = AutoTokenizer.from_pretrained("flax-community/gpt-neo-125M-code-clippy-dedup")
 prompt = """def greet(name):
   '''A function to greet user. Given a user name it should say hello'''