Update README.md
Browse files
README.md
CHANGED
@@ -29,7 +29,7 @@ Load the model in 8bit, then [run inference](https://huggingface.co/docs/transfo
|
|
29 |
```python
|
30 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
31 |
|
32 |
-
model_name = "ethzanalytics/stablelm-
|
33 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
34 |
|
35 |
model = AutoModelForCausalLM.from_pretrained(model_name, load_in_8bit=True)
|
|
|
29 |
```python
|
30 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
31 |
|
32 |
+
model_name = "ethzanalytics/stablelm-base-alpha-3b-sharded"
|
33 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
34 |
|
35 |
model = AutoModelForCausalLM.from_pretrained(model_name, load_in_8bit=True)
|