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

add IP_whitelisting to koboldai_settings.py

Browse Source
This commit is contained in:
YellowRoseCx
2023-04-05 21:27:59 -05:00
committed by GitHub
parent 71e5d23a5b
commit ac98cd6dd1
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
koboldai_settings.py
View File

@@ -19,6 +19,8 @@ import inspect
serverstarted = False serverstarted = False
queue = None queue = None
multi_story = False multi_story = False
global enable_whitelist
enable_whitelist = False
if importlib.util.find_spec("tortoise") is not None: if importlib.util.find_spec("tortoise") is not None:
from tortoise import api 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, 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]} "sampler_order": [6,0,1,2,3,4,5]}
def __init__(self, socketio, koboldai_vars): def __init__(self, socketio, koboldai_vars):
self.enable_whitelist = False
self._socketio = socketio self._socketio = socketio
self.reset_for_model_load() self.reset_for_model_load()
self.model = "" # Model ID string chosen at startup self.model = "" # Model ID string chosen at startup