Merge pull request #9 from henk717/overhaul

Overhaul
This commit is contained in:
ebolam 2022-06-08 18:44:21 -04:00 committed by GitHub
commit 001439be45
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 2 deletions

View File

@ -987,8 +987,14 @@ def general_startup():
if(modpath):
# 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
#==================================================================#
@ -1989,6 +1995,8 @@ def load_model(use_gpu=True, gpu_layers=None, initial_load=False, online_model="
if not vars.gamestarted:
setStartState()
sendsettings()
refresh_settings()
# Set up Flask routes