vsrinivas commited on
Commit
00ff2b2
·
verified ·
1 Parent(s): b33ad9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ def respond(message, chat_history, instruction, temperature=0.7):
51
  prompt = format_chat_prompt(message, chat_history, instruction)
52
  chat_history = chat_history + [[message, ""]]
53
  stream = generate_seqs(prompt = prompt,
54
- max_new_tokens=1024,
55
  stop_sequence=["\nUser:", "<|endoftext|>"],
56
  temperature=temperature).split('Assistant: ')[-1]
57
  #stop_sequence to not generate the user answer
 
51
  prompt = format_chat_prompt(message, chat_history, instruction)
52
  chat_history = chat_history + [[message, ""]]
53
  stream = generate_seqs(prompt = prompt,
54
+ max_new_tokens=8192,
55
  stop_sequence=["\nUser:", "<|endoftext|>"],
56
  temperature=temperature).split('Assistant: ')[-1]
57
  #stop_sequence to not generate the user answer