Spaces:
Runtime error
Runtime error
layerdiffusion
commited on
Commit
β’
016576a
1
Parent(s):
a1c7119
app.py
CHANGED
@@ -133,7 +133,7 @@ def chat_fn(message: str, history: list, seed:int, temperature: float, top_p: fl
|
|
133 |
|
134 |
def interactive_stopping_criteria(input_ids: torch.LongTensor, score: torch.FloatTensor, **kwargs) -> bool:
|
135 |
if getattr(streamer, 'user_interrupted', False):
|
136 |
-
print('User stopped generation')
|
137 |
return True
|
138 |
else:
|
139 |
return False
|
|
|
133 |
|
134 |
def interactive_stopping_criteria(input_ids: torch.LongTensor, score: torch.FloatTensor, **kwargs) -> bool:
|
135 |
if getattr(streamer, 'user_interrupted', False):
|
136 |
+
print('User stopped generation:', message)
|
137 |
return True
|
138 |
else:
|
139 |
return False
|