Update README.md
Browse files
README.md
CHANGED
@@ -57,7 +57,7 @@ while True:
|
|
57 |
# Данный пример промпта позволяет вести диалог и писать инструкции.
|
58 |
# prompt = '<SC6>Тебя зовут Анфиса. Тебе интересно машинное обучение.' + '\n'.join(dialog) + '\nТы ответил: <extra_id_0>'
|
59 |
# Второй пример - промпт просто для диалогов. В таком режиме не будет глюков, когда модель кидает кусок промпта в ответ.
|
60 |
-
prompt = '<
|
61 |
|
62 |
input_ids = t5_tokenizer(prompt, return_tensors='pt').input_ids
|
63 |
out_ids = t5_model.generate(input_ids=input_ids.to(device), generation_config=generation_config)
|
|
|
57 |
# Данный пример промпта позволяет вести диалог и писать инструкции.
|
58 |
# prompt = '<SC6>Тебя зовут Анфиса. Тебе интересно машинное обучение.' + '\n'.join(dialog) + '\nТы ответил: <extra_id_0>'
|
59 |
# Второй пример - промпт просто для диалогов. В таком режиме не будет глюков, когда модель кидает кусок промпта в ответ.
|
60 |
+
prompt = '<SC1>Тебя зовут Анфиса. Тебе интересно машинное обучение.' + '\n'.join(dialog) + '\nТы ответил: <extra_id_0>'
|
61 |
|
62 |
input_ids = t5_tokenizer(prompt, return_tensors='pt').input_ids
|
63 |
out_ids = t5_model.generate(input_ids=input_ids.to(device), generation_config=generation_config)
|