Spaces:
Running
Running
雷娃
commited on
Commit
·
fa749bc
1
Parent(s):
d287896
add interactive mode
Browse files
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(
|
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
|