Spaces:
Running
on
Zero
Running
on
Zero
layerdiffusion
commited on
Commit
•
f61fc21
1
Parent(s):
016576a
app.py
CHANGED
@@ -131,7 +131,7 @@ def chat_fn(message: str, history: list, seed:int, temperature: float, top_p: fl
|
|
131 |
|
132 |
streamer = TextIteratorStreamer(llm_tokenizer, timeout=10.0, skip_prompt=True, skip_special_tokens=True)
|
133 |
|
134 |
-
def interactive_stopping_criteria(
|
135 |
if getattr(streamer, 'user_interrupted', False):
|
136 |
print('User stopped generation:', message)
|
137 |
return True
|
|
|
131 |
|
132 |
streamer = TextIteratorStreamer(llm_tokenizer, timeout=10.0, skip_prompt=True, skip_special_tokens=True)
|
133 |
|
134 |
+
def interactive_stopping_criteria(*args, **kwargs) -> bool:
|
135 |
if getattr(streamer, 'user_interrupted', False):
|
136 |
print('User stopped generation:', message)
|
137 |
return True
|