Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ max_new_tokens = 1096,
|
|
| 12 |
threads = 3,
|
| 13 |
)
|
| 14 |
|
| 15 |
-
def stream(user_prompt):
|
| 16 |
system_prompt = 'Below is an instruction that describes a task. Write a response that appropriately completes the request.'
|
| 17 |
E_INST = "</s>"
|
| 18 |
user, assistant = "<|user|>", "<|assistant|>"
|
|
|
|
| 12 |
threads = 3,
|
| 13 |
)
|
| 14 |
|
| 15 |
+
def stream(user_prompt, mok):
|
| 16 |
system_prompt = 'Below is an instruction that describes a task. Write a response that appropriately completes the request.'
|
| 17 |
E_INST = "</s>"
|
| 18 |
user, assistant = "<|user|>", "<|assistant|>"
|