flagopen
/

CodeLlama-7b-Python-taco

Text Generation

competition-level_code_generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

rongaoli commited on May 23

Commit

01f64df

•

1 Parent(s): bfed216

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -38,8 +38,8 @@ You can use this model directly with a pipeline for text generation. This exampl
 ```py
 from transformers import AutoModelForCausalLM, AutoTokenizer, FlaxAutoModelForCausalLM
-model = AutoModelForCausalLM.from_pretrained("FlagOpen/CodeLlama-7b-Python-taco")
-tokenizer = AutoTokenizer.from_pretrained("FlagOpen/CodeLlama-7b-Python-taco")
 prompt = """
 A function to greet user. Given a user name it should say hello
 def greet(name):

 ```py
 from transformers import AutoModelForCausalLM, AutoTokenizer, FlaxAutoModelForCausalLM
+model = AutoModelForCausalLM.from_pretrained("flagopen/CodeLlama-7b-Python-taco")
+tokenizer = AutoTokenizer.from_pretrained("flagopen/CodeLlama-7b-Python-taco")
 prompt = """
 A function to greet user. Given a user name it should say hello
 def greet(name):