koziev ilya
commited on
Commit
•
d66a56e
1
Parent(s):
3126589
adding model.eval() to inference example
Browse files
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 |
# В конце добавляем символ "#"
|