AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Make exllama stoppable

Browse Source
This commit is contained in:
oobabooga 2023-06-16 22:03:23 -03:00
parent 7f06d551a3
commit b27f83c0e9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/exllama.py
View File

@ -79,7 +79,7 @@ class ExllamaModel:
for i in range(state['max_new_tokens']):
token = generator.gen_single_token()
yield (generator.tokenizer.decode(generator.sequence[0][initial_len:]))
if token.item() == generator.tokenizer.eos_token_id:
if token.item() == generator.tokenizer.eos_token_id or shared.stop_everything:
break
def encode(self, string, **kwargs):