From 45cd6b5de2f419548b821c22818ea376371fcf86 Mon Sep 17 00:00:00 2001
From: somebody <onesome01@protonmail.com>
Date: Mon, 27 Feb 2023 19:24:41 -0600
Subject: [PATCH] Model: Don't do breakmodel on colab

---
 model.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/model.py b/model.py
index ba12516a..44180341 100644
--- a/model.py
+++ b/model.py
@@ -44,9 +44,14 @@ from transformers import (
 )
 
 import utils
-import breakmodel
 import koboldai_settings
 
+try:
+    import breakmodel
+except ModuleNotFoundError as e:
+    if not utils.koboldai_vars.use_colab_tpu:
+        raise e
+
 HACK_currentmodel = None
 
 try: