koziev ilya commited on
Commit
d66a56e
1 Parent(s): 3126589

adding model.eval() to inference example

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -49,6 +49,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
49
  tokenizer.add_special_tokens({'bos_token': '<s>', 'eos_token': '</s>', 'pad_token': '<pad>'})
50
  model = AutoModelForCausalLM.from_pretrained(model_name)
51
  model.to(device)
 
52
 
53
  # На вход модели подаем последние 2-3 реплики диалога. Каждая реплика на отдельной строке, начинается с символа "-"
54
  # В конце добавляем символ "#"
 
49
  tokenizer.add_special_tokens({'bos_token': '<s>', 'eos_token': '</s>', 'pad_token': '<pad>'})
50
  model = AutoModelForCausalLM.from_pretrained(model_name)
51
  model.to(device)
52
+ model.eval()
53
 
54
  # На вход модели подаем последние 2-3 реплики диалога. Каждая реплика на отдельной строке, начинается с символа "-"
55
  # В конце добавляем символ "#"