Model: Don't do breakmodel on colab

This commit is contained in:
somebody
2023-02-27 19:24:41 -06:00
parent ed83362dee
commit 45cd6b5de2

View File

@@ -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: