From 6fa50f78bf5f19eab39354b73fda25ae13c7e074 Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Mon, 17 Feb 2025 00:48:34 +0800
Subject: [PATCH] allow kcppt for config switching
---
koboldcpp.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
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