PEFT
English
medical
Tonic commited on
Commit
0a89b4a
1 Parent(s): dec0276

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -103,7 +103,7 @@ class ChatBot:
103
  user_input_ids = tokenizer.encode(input + tokenizer.eos_token, return_tensors="pt")
104
 
105
  # Concatenate the user input with chat history
106
- if self.history.numel() > 0:
107
  chat_history_ids = torch.cat([self.history, user_input_ids], dim=-1)
108
  else:
109
  chat_history_ids = user_input_ids
 
103
  user_input_ids = tokenizer.encode(input + tokenizer.eos_token, return_tensors="pt")
104
 
105
  # Concatenate the user input with chat history
106
+ if len(self.history) > 0:
107
  chat_history_ids = torch.cat([self.history, user_input_ids], dim=-1)
108
  else:
109
  chat_history_ids = user_input_ids