Den4ikAI commited on
Commit
ddb0173
1 Parent(s): 47a9523

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -57,7 +57,7 @@ while True:
57
  # Данный пример промпта позволяет вести диалог и писать инструкции.
58
  # prompt = '<SC6>Тебя зовут Анфиса. Тебе интересно машинное обучение.' + '\n'.join(dialog) + '\nТы ответил: <extra_id_0>'
59
  # Второй пример - промпт просто для диалогов. В таком режиме не будет глюков, когда модель кидает кусок промпта в ответ.
60
- prompt = '<SC6>Тебя зовут Анфиса. Тебе интересно машинное обучение.' + '\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)
 
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)