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

Added ability to use env variables instead of argparse (command argument = docker env variable)

Browse Source
This commit is contained in:
ebolam
2022-07-01 16:09:13 -04:00
parent 40bcf893d5
commit ca07fdbe44
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
aiserver.py
View File

@@ -908,6 +908,11 @@ def general_startup(override_args=None):
else: else:
args = parser.parse_args() args = parser.parse_args()
for arg in vars(args):
if arg in os.environ:
setattr(args, arg, os.environ[arg])
koboldai_vars.model = args.model; koboldai_vars.model = args.model;
koboldai_vars.revision = args.revision koboldai_vars.revision = args.revision

4
koboldai_settings.py
View File

@@ -308,7 +308,7 @@ class story_settings(settings):
self.notes = "" #Notes for the story. Does nothing but save self.notes = "" #Notes for the story. Does nothing but save
def reset(self): def reset(self):
self.__init(self.socketio, tokenizer=self.tokenizer) self.__init__(self.socketio, tokenizer=self.tokenizer)
def __setattr__(self, name, value): def __setattr__(self, name, value):
new_variable = name not in self.__dict__ new_variable = name not in self.__dict__
@@ -543,7 +543,7 @@ class KoboldStoryRegister(object):
else: else:
self.actions[self.action_count]['Selected Text Length'] = None self.actions[self.action_count]['Selected Text Length'] = None
process_variable_changes(self.socketio, "actions", "Selected Text", {"id": self.action_count, "text": text}, None) process_variable_changes(self.socketio, "actions", "Selected Text", {"id": self.action_count, "text": text}, None)
process_variable_changes(self.socketio, "actions", 'Selected Text Length', {"id": self.action_count, 'length': self.actions[self.action_count]['Selected Text Length']}, {"id": action_step, 'length': 0}) process_variable_changes(self.socketio, "actions", 'Selected Text Length', {"id": self.action_count, 'length': self.actions[self.action_count]['Selected Text Length']}, {"id": self.action_count, 'length': 0})
self.set_game_saved() self.set_game_saved()
def append_options(self, option_list): def append_options(self, option_list):