雷娃 commited on
Commit
fa749bc
·
1 Parent(s): d287896

add interactive mode

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ def chat(user_input, max_new_tokens=512):
36
  thread = Thread(target=generate)
37
  thread.start()
38
 
39
- start_idx = len(inputs)
40
  generated_text = ""
41
  for new_text in streamer:
42
  generated_text += new_text
 
36
  thread = Thread(target=generate)
37
  thread.start()
38
 
39
+ start_idx = len("SYSTEM") + len(messages[0]["content"]) + len("HUMAN") + len(user_input) + len("ASSISTANT")
40
  generated_text = ""
41
  for new_text in streamer:
42
  generated_text += new_text