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

Model: More tpu fixes

Browse Source
This commit is contained in:
somebody
2023-02-27 19:29:03 -06:00
parent 45cd6b5de2
commit bd3bbdaad8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@@ -66,7 +66,6 @@ import lupa
# KoboldAI # KoboldAI
import fileops import fileops
import gensettings import gensettings
import breakmodel
from utils import debounce from utils import debounce
import utils import utils
import koboldai_settings import koboldai_settings
@@ -7802,6 +7801,7 @@ def UI_2_get_applicable_genres():
@socketio.on("create_new_softprompt") @socketio.on("create_new_softprompt")
@logger.catch @logger.catch
def UI_2_create_new_softprompt(data): def UI_2_create_new_softprompt(data):
import breakmodel
logger.info("Soft Prompt Dataset: {}".format(data)) logger.info("Soft Prompt Dataset: {}".format(data))
from prompt_tuner import BasicTrainer from prompt_tuner import BasicTrainer
trainer = BasicTrainer(None, quiet=koboldai_vars.quiet) trainer = BasicTrainer(None, quiet=koboldai_vars.quiet)