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 |
)
|