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

Model: Don't do breakmodel on colab

Browse Source
This commit is contained in:
somebody
2023-02-27 19:24:41 -06:00
parent ed83362dee
commit 45cd6b5de2
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
model.py
View File

@@ -44,9 +44,14 @@ from transformers import (
) )
import utils import utils
import breakmodel
import koboldai_settings import koboldai_settings
try:
import breakmodel
except ModuleNotFoundError as e:
if not utils.koboldai_vars.use_colab_tpu:
raise e
HACK_currentmodel = None HACK_currentmodel = None
try: try: