Explorar o código

Disable the hijack after using it

oobabooga %!s(int64=2) %!d(string=hai) anos
pai
achega
85f914b9b9
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  1. 1 0
      modules/chat.py

+ 1 - 0
modules/chat.py

@@ -96,6 +96,7 @@ def chatbot_wrapper(text, max_new_tokens, do_sample, temperature, top_p, typical
     custom_prompt_generator = None
     for extension, _ in extensions_module.iterator():
         if hasattr(extension, 'input_hijack') and extension.input_hijack['state'] == True:
+            extension.input_hijack['state'] = False
             text, visible_text = extension.input_hijack['value']
         if custom_prompt_generator is None and hasattr(extension, 'custom_prompt_generator'):
             custom_prompt_generator = extension.custom_prompt_generator