add IP_whitelisting to koboldai_settings.py

This commit is contained in:
YellowRoseCx
2023-04-05 21:27:59 -05:00
committed by GitHub
parent 71e5d23a5b
commit ac98cd6dd1

View File

@@ -19,6 +19,8 @@ import inspect
serverstarted = False
queue = None
multi_story = False
global enable_whitelist
enable_whitelist = False
if importlib.util.find_spec("tortoise") is not None:
from tortoise import api
@@ -654,6 +656,7 @@ class model_settings(settings):
default_settings = {"rep_pen" : 1.1, "rep_pen_slope": 0.7, "rep_pen_range": 1024, "temp": 0.5, "top_p": 0.9, "top_k": 0, "top_a": 0.0, "tfs": 1.0, "typical": 1.0,
"sampler_order": [6,0,1,2,3,4,5]}
def __init__(self, socketio, koboldai_vars):
self.enable_whitelist = False
self._socketio = socketio
self.reset_for_model_load()
self.model = "" # Model ID string chosen at startup