Text Generation
Russian
conversational
IlyaGusev commited on
Commit
271e752
1 Parent(s): d7c13be

Update README.md

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