flagopen
/

starcoderbase-1b-taco

Text Generation

competition-level_code_generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

rongaoli commited on May 23

Commit

996d653

•

1 Parent(s): 49a700e

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/starcoderbase-1b-taco")
-tokenizer = AutoTokenizer.from_pretrained("FlagOpen/starcoderbase-1b-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/starcoderbase-1b-taco")
+tokenizer = AutoTokenizer.from_pretrained("flagopen/starcoderbase-1b-taco")
 prompt = """
 A function to greet user. Given a user name it should say hello
 def greet(name):