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

Warning polish

Browse Source
This commit is contained in:
Henk
2023-05-11 15:13:59 +02:00
parent e932364a1e
commit edd9c7d782
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
koboldai_settings.py
View File

@@ -1129,7 +1129,7 @@ class story_settings(settings):
class user_settings(settings): class user_settings(settings):
local_only_variables = ['importjs'] local_only_variables = ['importjs']
no_save_variables = ['importnum', 'importjs', 'loadselect', 'spselect', 'svowname', 'saveow', 'laststory', 'sid', "revision"] no_save_variables = ['importnum', 'importjs', 'loadselect', 'spselect', 'svowname', 'saveow', 'laststory', 'sid', "revision", "model_selected"]
settings_name = "user" settings_name = "user"
def __init__(self, socketio): def __init__(self, socketio):
self._socketio = socketio self._socketio = socketio
@@ -1185,6 +1185,7 @@ class user_settings(settings):
self.horde_api_key = "0000000000" self.horde_api_key = "0000000000"
self.horde_worker_name = "My Awesome Instance" self.horde_worker_name = "My Awesome Instance"
self.horde_url = "https://horde.koboldai.net" self.horde_url = "https://horde.koboldai.net"
self.model_selected = ""
def __setattr__(self, name, value): def __setattr__(self, name, value):
new_variable = name not in self.__dict__ new_variable = name not in self.__dict__

2
modeling/inference_model.py
View File

@@ -218,7 +218,7 @@ class InferenceModel:
try: try:
return GenericTokenizer(try_get_tokenizer()) return GenericTokenizer(try_get_tokenizer())
except Exception as e: except Exception as e:
logger.warning(f"Tokenizer falling back due to {e}") logger.warning(f"Tokenizer falling back due to {e} (This can be normal behavior for some architectures that lack a slow tokenizer such as NeoX)")
# If we error on each attempt, raise the last one # If we error on each attempt, raise the last one
if i == len(suppliers) - 1: if i == len(suppliers) - 1:
raise raise