IlyaGusev commited on
Commit
cd94e95
1 Parent(s): 1a5ce42

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -66,7 +66,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
66
  model.eval()
67
 
68
  generation_config = GenerationConfig.from_pretrained(MODEL_NAME)
69
- data = tokenizer(["Вопрос: Почему трава зеленая?\n\nВыход:"], return_tensors="pt")
70
  data = {k: v.to(model.device) for k, v in data.items() if k in ("input_ids", "attention_mask")}
71
  output_ids = model.generate(**data, generation_config=generation_config)[0]
72
  print(tokenizer.decode(output_ids, skip_special_tokens=True))
 
66
  model.eval()
67
 
68
  generation_config = GenerationConfig.from_pretrained(MODEL_NAME)
69
+ data = tokenizer(["### Задание: Почему трава зеленая?\n### Ответ: "], return_tensors="pt")
70
  data = {k: v.to(model.device) for k, v in data.items() if k in ("input_ids", "attention_mask")}
71
  output_ids = model.generate(**data, generation_config=generation_config)[0]
72
  print(tokenizer.decode(output_ids, skip_special_tokens=True))