rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Fix for new prompt bug

Browse Source
This commit is contained in:
ebolam
2022-08-28 11:03:05 -04:00
parent 33ab77736b
commit 3d550f1c72
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
aiserver.py
View File

@@ -40,6 +40,7 @@ import traceback
import inspect import inspect
import warnings import warnings
import multiprocessing import multiprocessing
import copy
from collections.abc import Iterable from collections.abc import Iterable
from collections import OrderedDict from collections import OrderedDict
from typing import Any, Callable, TypeVar, Tuple, Union, Dict, Set, List, Optional, Type from typing import Any, Callable, TypeVar, Tuple, Union, Dict, Set, List, Optional, Type
@@ -4587,7 +4588,7 @@ def _generate(txt, minimum, maximum, found_entries):
koboldai_vars._actions = koboldai_vars.actions koboldai_vars._actions = koboldai_vars.actions
koboldai_vars._prompt = koboldai_vars.prompt koboldai_vars._prompt = koboldai_vars.prompt
if(koboldai_vars.dynamicscan): if(koboldai_vars.dynamicscan):
koboldai_vars._actions = koboldai_vars._actions.copy() koboldai_vars._actions = copy.deepcopy(koboldai_vars._actions)
with torch.no_grad(): with torch.no_grad():
already_generated = 0 already_generated = 0
@@ -7045,7 +7046,6 @@ def UI_2_Set_Selected_Text(data):
#==================================================================# #==================================================================#
@socketio.on('Use Option Text') @socketio.on('Use Option Text')
def UI_2_Use_Option_Text(data): def UI_2_Use_Option_Text(data):
print("Using Option Text: {}".format(data))
if koboldai_vars.prompt == "": if koboldai_vars.prompt == "":
koboldai_vars.prompt = koboldai_vars.actions.get_current_options()[int(data['option'])]['text'] koboldai_vars.prompt = koboldai_vars.actions.get_current_options()[int(data['option'])]['text']
koboldai_vars.actions.clear_unused_options() koboldai_vars.actions.clear_unused_options()