EpistemeAI
/

Athena-codegemma-2-2b-it

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

legolasyiu commited on Sep 5

Commit

63142c8

•

1 Parent(s): a1d75a1

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -104,7 +104,7 @@ You can also use `float32` if you skip the dtype, but no precision increase will
 from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b-it")
 model = AutoModelForCausalLM.from_pretrained(
-    "google/gemma-2-2b-it",
     device_map="auto",
 )
 input_text = "Write me a poem about Machine Learning."

 from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b-it")
 model = AutoModelForCausalLM.from_pretrained(
+    "EpistemeAI/Athena-gemma-2-2b-it",
     device_map="auto",
 )
 input_text = "Write me a poem about Machine Learning."