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

Merge pull request #9 from henk717/overhaul

Browse Source 
Overhaul
This commit is contained in:
ebolam
2022-06-08 18:44:21 -04:00
committed by GitHub
parent 622a3fc8db 1a46d97ad5
commit 001439be45
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
aiserver.py
View File

@ -987,8 +987,14 @@ def general_startup():
if(modpath): if(modpath):
# Save directory to vars # Save directory to vars
vars.model = modpath vars.model = "NeoCustom"
vars.custmodpth = modpath
elif args.model:
print("Welcome to KoboldAI!\nYou have selected the following Model:", vars.model)
if args.path:
print("You have selected the following path for your Model :", args.path)
vars.custmodpth = args.path;
vars.colaburl = args.path + "/request"; # Lets just use the same parameter to keep it simple
#==================================================================# #==================================================================#
# Load Model # Load Model
#==================================================================# #==================================================================#
@ -1989,6 +1995,8 @@ def load_model(use_gpu=True, gpu_layers=None, initial_load=False, online_model="
if not vars.gamestarted: if not vars.gamestarted:
setStartState() setStartState()
sendsettings()
refresh_settings()
# Set up Flask routes # Set up Flask routes