vsrinivas commited on
Commit
5a8970f
·
verified ·
1 Parent(s): da70ed8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -89,6 +89,7 @@ def format_chat_prompt(message, chat_history, instruction):
89
  def generate_seqs(prompt, max_new_tokens=None, stop_sequence=None, temperature=None):
90
  output = pipeline(prompt,
91
  max_length=200,
 
92
  max_new_tokens = max_new_tokens,
93
  stop_sequence = stop_sequence,
94
  temperature=temperature,
 
89
  def generate_seqs(prompt, max_new_tokens=None, stop_sequence=None, temperature=None):
90
  output = pipeline(prompt,
91
  max_length=200,
92
+ truncation=True,
93
  max_new_tokens = max_new_tokens,
94
  stop_sequence = stop_sequence,
95
  temperature=temperature,