gathnex commited on
Commit
00b34c0
1 Parent(s): 51f2dab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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|>"