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 env variables

Browse Source
This commit is contained in:
ebolam
2022-07-01 16:49:47 -04:00
parent 18dd97de6b
commit 73ad11c6d7
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
aiserver.py
View File

@@ -909,8 +909,15 @@ def general_startup(override_args=None):
args = parser.parse_args()
for arg in vars(args):
if arg in os.environ:
setattr(args, arg, os.environ[arg])
if arg == "path":
if "model_path" in os.environ:
setattr(args, arg, os.environ["model_path"])
else:
if arg in os.environ:
if type(getattr(args, arg)) == bool:
setattr(args, arg, bool(os.environ[arg]))
else:
setattr(args, arg, os.environ[arg])
koboldai_vars.model = args.model;