Sharathhebbar24 commited on
Commit
f1014d7
·
verified ·
1 Parent(s): 11bdb29

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -35,7 +35,7 @@ prompt.
35
 
36
  ```python
37
  >>> from transformers import AutoTokenizer, AutoModelForCausalLM
38
- >>> model_name = "Sharathhebbar24/Instruct_GPT_v1"
39
  >>> model = AutoModelForCausalLM.from_pretrained(model_name)
40
  >>> tokenizer = AutoTokenizer.from_pretrained(model_name)
41
  >>> def generate_text(prompt):
 
35
 
36
  ```python
37
  >>> from transformers import AutoTokenizer, AutoModelForCausalLM
38
+ >>> model_name = "Sharathhebbar24/SSH_355M"
39
  >>> model = AutoModelForCausalLM.from_pretrained(model_name)
40
  >>> tokenizer = AutoTokenizer.from_pretrained(model_name)
41
  >>> def generate_text(prompt):