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: Lazyload backends

Browse Source
This commit is contained in:
somebody
2023-03-13 20:29:29 -05:00
parent adc11fdbc9
commit b93c339145
5 changed files with 15 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modeling/warpers.py
View File

@@ -42,13 +42,10 @@ import utils
import torch
import numpy as np
try:
if utils.koboldai_vars.use_colab_tpu:
import jax
import jax.numpy as jnp
import tpu_mtj_backend
except ImportError as e:
if utils.koboldai_vars.use_colab_tpu:
raise e
def update_settings():