Update README.md
Browse files
README.md
CHANGED
@@ -31,12 +31,10 @@ class Conversation:
|
|
31 |
self,
|
32 |
message_template=DEFAULT_MESSAGE_TEMPLATE,
|
33 |
system_prompt=DEFAULT_SYSTEM_PROMPT,
|
34 |
-
role_mapping=None,
|
35 |
start_token_id=1,
|
36 |
bot_token_id=9225
|
37 |
):
|
38 |
self.message_template = message_template
|
39 |
-
self.role_mapping = role_mapping or {}
|
40 |
self.start_token_id = start_token_id
|
41 |
self.bot_token_id = bot_token_id
|
42 |
self.messages = [{
|
|
|
31 |
self,
|
32 |
message_template=DEFAULT_MESSAGE_TEMPLATE,
|
33 |
system_prompt=DEFAULT_SYSTEM_PROMPT,
|
|
|
34 |
start_token_id=1,
|
35 |
bot_token_id=9225
|
36 |
):
|
37 |
self.message_template = message_template
|
|
|
38 |
self.start_token_id = start_token_id
|
39 |
self.bot_token_id = bot_token_id
|
40 |
self.messages = [{
|