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

Debug

Browse Source
This commit is contained in:
ebolam
2022-10-24 19:28:56 -04:00
parent f0dfde9a64
commit 693e7500cf
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tpu_mtj_backend.py
View File

@@ -1205,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) print("Connecting to your Colab instance's TPU", flush=True)
old_ai_busy = koboldai_vars.aibusy old_ai_busy = koboldai_vars.aibusy
koboldai_vars.status_message = "Connecting to TPU" koboldai_vars.status_message = "Connecting to TPU"
spinner = threading.Thread((target=show_spinner, args=()) spinner = threading.Thread(target=show_spinner, args=())
spinner.start() spinner.start()
if os.environ.get('COLAB_TPU_ADDR', '') != '': if os.environ.get('COLAB_TPU_ADDR', '') != '':
tpu_address = os.environ['COLAB_TPU_ADDR'] # Colab tpu_address = os.environ['COLAB_TPU_ADDR'] # Colab