Den4ikAI commited on
Commit
89e379c
1 Parent(s): 66ae9b7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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(sample, return_tensors="pt").to(model.device)
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