From bd3bbdaad83235d9ba6a963f4810176b497315ab Mon Sep 17 00:00:00 2001
From: somebody <onesome01@protonmail.com>
Date: Mon, 27 Feb 2023 19:29:03 -0600
Subject: [PATCH] Model: More tpu fixes

---
 aiserver.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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