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: