Commit
•
2c45a92
1
Parent(s):
21d629b
Update README.md
Browse files
README.md
CHANGED
|
@@ -53,7 +53,7 @@ language:
|
|
| 53 |
pipeline_tag: text-generation
|
| 54 |
---
|
| 55 |
|
| 56 |
-
# Gistral
|
| 57 |
|
| 58 |
|
| 59 |
|
|
@@ -76,7 +76,7 @@ Use the code below to get started with the model.
|
|
| 76 |
|
| 77 |
```py
|
| 78 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 79 |
-
model_id = "ehristoforu/Gistral-16B
|
| 80 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 81 |
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
|
| 82 |
messages = [
|
|
|
|
| 53 |
pipeline_tag: text-generation
|
| 54 |
---
|
| 55 |
|
| 56 |
+
# Gistral 16B (Mistral from 7B to 16B)
|
| 57 |
|
| 58 |
|
| 59 |
|
|
|
|
| 76 |
|
| 77 |
```py
|
| 78 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 79 |
+
model_id = "ehristoforu/Gistral-16B"
|
| 80 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 81 |
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
|
| 82 |
messages = [
|