Merge pull request #16 from VE-FORBRYDERNE/staging

Fix compatibility with UI2
VE-FORBRYDERNE · Dec 26, 2022 · 751fb40 · 751fb40
2 parents 3dd0bf5 + cde795f
commit 751fb40
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/mtj_softtuner/kobold b/mtj_softtuner/kobold
diff --git a/mtj_softtuner/patch.py b/mtj_softtuner/patch.py
@@ -76,12 +76,12 @@ def decorated(*args, **kwargs):
 
  # Allows mtj-softtuner to use aria2 to download models
 
- old_vars = utils.vars
+ old_vars = utils.koboldai_vars
 
  class Vars:
  aria2_port = 6799
 
- utils.vars = Vars
+ utils.koboldai_vars = Vars
 
  logger = logging.getLogger("urllib3")
  old_level = logger.level
@@ -136,7 +136,7 @@ def new_get_checkpoint_shard_files(
  if JAX13:
  jax.host_count = old_jax_host_count
  jax.host_id = old_jax_host_id
- utils.vars = old_vars
+ utils.koboldai_vars = old_vars
  logger.setLevel(old_level)
  transformers.PreTrainedModel.from_pretrained = old_from_pretrained
  if hasattr(transformers.modeling_utils, "get_checkpoint_shard_files"):