瀏覽代碼

Change the default names (#518)

* Update shared.py

* Update settings-template.json
oobabooga 2 年之前
父節點
當前提交
c5ebcc5f7e
共有 2 個文件被更改,包括 6 次插入6 次删除
  1. 3 3
      modules/shared.py
  2. 3 3
      settings-template.json

+ 3 - 3
modules/shared.py

@@ -27,9 +27,9 @@ settings = {
     'max_new_tokens': 200,
     'max_new_tokens_min': 1,
     'max_new_tokens_max': 2000,
-    'name1': 'Person 1',
-    'name2': 'Person 2',
-    'context': 'This is a conversation between two people.',
+    'name1': 'You',
+    'name2': 'Assistant',
+    'context': 'This is a conversation with your Assistant. The Assistant is very helpful and is eager to chat with you and answer your questions.',
     'stop_at_newline': False,
     'chat_prompt_size': 2048,
     'chat_prompt_size_min': 0,

+ 3 - 3
settings-template.json

@@ -2,9 +2,9 @@
     "max_new_tokens": 200,
     "max_new_tokens_min": 1,
     "max_new_tokens_max": 2000,
-    "name1": "Person 1",
-    "name2": "Person 2",
-    "context": "This is a conversation between two people.",
+    "name1": "You",
+    "name2": "Assistant",
+    "context": "This is a conversation with your Assistant. The Assistant is very helpful and is eager to chat with you and answer your questions.",
     "stop_at_newline": false,
     "chat_prompt_size": 2048,
     "chat_prompt_size_min": 0,