Update README.md
Browse files
README.md
CHANGED
@@ -23,7 +23,7 @@ from transformers import GenerationConfig
|
|
23 |
|
24 |
generation_config = GenerationConfig.from_pretrained("Den4ikAI/FRED-T5-XL_instructor")
|
25 |
def generate(prompt):
|
26 |
-
data = tokenizer(
|
27 |
output_ids = model.generate(
|
28 |
**data,
|
29 |
generation_config=generation_config
|
|
|
23 |
|
24 |
generation_config = GenerationConfig.from_pretrained("Den4ikAI/FRED-T5-XL_instructor")
|
25 |
def generate(prompt):
|
26 |
+
data = tokenizer(f"<SC6>Человек: {prompt}\nБот: <extra_id_0>", return_tensors="pt").to(model.device)
|
27 |
output_ids = model.generate(
|
28 |
**data,
|
29 |
generation_config=generation_config
|