Spaces:
Paused
Paused
Update app.py
Browse files
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
|