Tonic commited on
Commit
fda9ddb
·
1 Parent(s): 67c4753

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -51,7 +51,6 @@ class FalconChatBot:
51
  return filtered_history
52
 
53
  def predict(self, user_message, assistant_message, history, temperature=0.4, max_new_tokens=700, top_p=0.99, repetition_penalty=1.9):
54
- input_ids = input_ids.to(device)
55
  # Process the history to remove special commands
56
  processed_history = self.process_history(history)
57
  # Combine the user and assistant messages into a conversation
 
51
  return filtered_history
52
 
53
  def predict(self, user_message, assistant_message, history, temperature=0.4, max_new_tokens=700, top_p=0.99, repetition_penalty=1.9):
 
54
  # Process the history to remove special commands
55
  processed_history = self.process_history(history)
56
  # Combine the user and assistant messages into a conversation