Sharathhebbar24
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -35,7 +35,7 @@ prompt.
|
|
35 |
|
36 |
```python
|
37 |
>>> from transformers import AutoTokenizer, AutoModelForCausalLM
|
38 |
-
>>> model_name = "Sharathhebbar24/
|
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):
|