From f0dfde9a6463a72af9d0e74895b69e2dbfdfbf7f Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Mon, 24 Oct 2022 19:27:48 -0400
Subject: [PATCH] Colab debug

---
 tpu_mtj_backend.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tpu_mtj_backend.py b/tpu_mtj_backend.py
index 672006a2..4242186a 100644
--- a/tpu_mtj_backend.py
+++ b/tpu_mtj_backend.py
@@ -30,6 +30,7 @@ SOFTWARE.
 import utils
 
 import multiprocessing
+import threading
 from typing import Any, Callable, Dict, List, NamedTuple, Optional, Tuple, TypeVar
 import progressbar
 import time
@@ -114,7 +115,7 @@ def compiling_callback() -> None:
     pass
 
 
-def show_spinner(socketio):
+def show_spinner():
     bar = progressbar.ProgressBar(max_value=progressbar.UnknownLength, widgets=[progressbar.Timer(), '  ', progressbar.BouncingBar(left='[', right=']', marker='█')])
     i = 0
     while True:
@@ -1204,7 +1205,7 @@ def load_model(path: str, driver_version="tpu_driver0.1_dev20210607", hf_checkpo
     print("Connecting to your Colab instance's TPU", flush=True)
     old_ai_busy = koboldai_vars.aibusy
     koboldai_vars.status_message = "Connecting to TPU"
-    spinner = multiprocessing.Process(target=show_spinner, args=(socketio))
+    spinner = threading.Thread((target=show_spinner, args=())
     spinner.start()
     if os.environ.get('COLAB_TPU_ADDR', '') != '':
         tpu_address = os.environ['COLAB_TPU_ADDR']  # Colab