IlyaGusev commited on
Commit
63f5b25
1 Parent(s): 755e7be

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -2
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 = [{