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

Env Fix

Browse Source
This commit is contained in:
ebolam
2022-07-01 17:24:06 -04:00
parent 6e841b87eb
commit 0161966cea
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
aiserver.py
View File

@@ -908,19 +908,21 @@ def general_startup(override_args=None):
else: else:
args = parser.parse_args() args = parser.parse_args()
for arg in vars(args): temp = [x for x in vars(args)]
for arg in temp:
if arg == "path": if arg == "path":
if "model_path" in os.environ: if "model_path" in os.environ:
setattr(args, arg, os.environ["model_path"]) setattr(args, arg, os.environ["model_path"])
else: else:
if arg in os.environ: if arg in os.environ:
if type(getattr(args, arg)) == bool: if isinstance(getattr(args, arg), bool):
if bool(os.environ[arg]): if os.environ[arg].lower() == "true":
setattr(args, arg, bool(os.environ[arg])) setattr(args, arg, True)
else: else:
delattr(args, arg) setattr(args, arg, False)
else: else:
setattr(args, arg, os.environ[arg]) setattr(args, arg, os.environ[arg])
koboldai_vars.model = args.model; koboldai_vars.model = args.model;