Hugging Face's logo

ed001
/
datagemma-2b

Text Generation
Transformers
Safetensors
English
gemma
code
data science
text-generation-inference
Inference Endpoints
Model card Files Community
ed001 commited on
Commit
41bab0d
·
verified ·
1 Parent(s): 73a8023

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -25,7 +25,7 @@ model = AutoModelForCausalLM.from_pretrained(
25
  ).cuda()
26
 
27
  # Reload tokenizer to save it
28
- tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
29
  tokenizer.padding_side = "right"
30
 
31
  prompt_template = "### Question: {}\n ### Answer: "
 
25
  ).cuda()
26
 
27
  # Reload tokenizer to save it
28
+ tokenizer = AutoTokenizer.from_pretrained("ed001/datagemma-2b", trust_remote_code=True)
29
  tokenizer.padding_side = "right"
30
 
31
  prompt_template = "### Question: {}\n ### Answer: "