Update README.md
Browse files
README.md
CHANGED
|
@@ -9,7 +9,7 @@ language:
|
|
| 9 |
library_name: transformers
|
| 10 |
---
|
| 11 |
|
| 12 |
-
#
|
| 13 |
|
| 14 |
This is the 1B OPT model additionally pretrained on Slovene data. The model was created as a part of project Povejmo: https://www.cjvt.si/povejmo/.
|
| 15 |
|
|
@@ -41,11 +41,12 @@ The inference can be done using the following snippet of code:
|
|
| 41 |
```python
|
| 42 |
from transformers import AutoTokenizer, pipeline
|
| 43 |
|
| 44 |
-
|
|
|
|
| 45 |
|
| 46 |
pline = pipeline(
|
| 47 |
"text-generation",
|
| 48 |
-
model=
|
| 49 |
tokenizer=tokenizer,
|
| 50 |
device_map="auto"
|
| 51 |
)
|
|
|
|
| 9 |
library_name: transformers
|
| 10 |
---
|
| 11 |
|
| 12 |
+
# OPT_GaMS 1B
|
| 13 |
|
| 14 |
This is the 1B OPT model additionally pretrained on Slovene data. The model was created as a part of project Povejmo: https://www.cjvt.si/povejmo/.
|
| 15 |
|
|
|
|
| 41 |
```python
|
| 42 |
from transformers import AutoTokenizer, pipeline
|
| 43 |
|
| 44 |
+
model_id = "cjvt/OPT_GaMS-1B"
|
| 45 |
+
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 46 |
|
| 47 |
pline = pipeline(
|
| 48 |
"text-generation",
|
| 49 |
+
model=model_id,
|
| 50 |
tokenizer=tokenizer,
|
| 51 |
device_map="auto"
|
| 52 |
)
|