diff --git a/koboldcpp.py b/koboldcpp.py index b08328d8d..e82378feb 100644 --- a/koboldcpp.py +++ b/koboldcpp.py @@ -2408,7 +2408,7 @@ Enter Prompt:
opts = [] if args.admin and args.admindir and os.path.exists(args.admindir) and self.check_header_password(args.adminpassword): dirpath = os.path.abspath(args.admindir) - opts = [f for f in sorted(os.listdir(dirpath)) if f.endswith(".kcpps") and os.path.isfile(os.path.join(dirpath, f))] + opts = [f for f in sorted(os.listdir(dirpath)) if (f.endswith(".kcpps") or f.endswith(".kcppt")) and os.path.isfile(os.path.join(dirpath, f))] response_body = (json.dumps(opts).encode()) elif self.path.endswith(('/api/extra/perf')): @@ -2780,7 +2780,7 @@ Enter Prompt:
if targetfile and targetfile!="": dirpath = os.path.abspath(args.admindir) targetfilepath = os.path.join(dirpath, targetfile) - opts = [f for f in os.listdir(dirpath) if f.endswith(".kcpps") and os.path.isfile(os.path.join(dirpath, f))] + opts = [f for f in os.listdir(dirpath) if (f.endswith(".kcpps") or f.endswith(".kcppt")) and os.path.isfile(os.path.join(dirpath, f))] if targetfile in opts and os.path.exists(targetfilepath): print(f"Admin: Received request to reload config to {targetfile}") global_memory["restart_target"] = targetfile