iot
/

Gemma_model_fine_tune_custom_Data

Text Generation

text-generation-inference

Inference Endpoints

4-bit precision

Model card Files Files and versions Metrics Training metrics Community

iot commited on May 24

Commit

cba7ae9

•

1 Parent(s): b177b04

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -21,8 +21,8 @@ You can use this model, Below is an example of how to load and use the model:
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
-tokenizer = AutoTokenizer.from_pretrained("username/Gemma_model_fine_tune_custom_Data")
-model = AutoModelForCausalLM.from_pretrained("username/Gemma_model_fine_tune_custom_Data")
 input_text = "Your input text here"
 inputs = tokenizer(input_text, return_tensors="pt")

 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained("iot/Gemma_model_fine_tune_custom_Data")
+model = AutoModelForCausalLM.from_pretrained("iot/Gemma_model_fine_tune_custom_Data")
 input_text = "Your input text here"
 inputs = tokenizer(input_text, return_tensors="pt")