Update README.md
Browse files
README.md
CHANGED
@@ -12,8 +12,8 @@ import torch
|
|
12 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
13 |
from transformers import GenerationConfig
|
14 |
|
15 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
16 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("
|
17 |
model.eval()
|
18 |
|
19 |
def generate(prompt):
|
|
|
12 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
13 |
from transformers import GenerationConfig
|
14 |
|
15 |
+
tokenizer = AutoTokenizer.from_pretrained("SiberiaSoft/SiberianFredT5-instructor")
|
16 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("SiberiaSoft/SiberianFredT5-instructor")
|
17 |
model.eval()
|
18 |
|
19 |
def generate(prompt):
|