From b5272ea607ad38e162b4625893ee491900305342 Mon Sep 17 00:00:00 2001 From: ebolam Date: Wed, 24 May 2023 20:15:11 -0400 Subject: [PATCH] Whoops --- tpu_mtj_backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tpu_mtj_backend.py b/tpu_mtj_backend.py index bf08f745..df37e0be 100644 --- a/tpu_mtj_backend.py +++ b/tpu_mtj_backend.py @@ -943,7 +943,7 @@ def read_neox_checkpoint(state, path, config, checkpoint_shards=2): import koboldai_settings -def load_model(path: str, model_type: str, badwordsids=koboldai_settings.badwordsids_default driver_version="tpu_driver_20221109", hf_checkpoint=False, socketio_queue=None, initial_load=False, logger=None, **kwargs) -> None: +def load_model(path: str, model_type: str, badwordsids=koboldai_settings.badwordsids_default, driver_version="tpu_driver_20221109", hf_checkpoint=False, socketio_queue=None, initial_load=False, logger=None, **kwargs) -> None: global thread_resources_env, seq, tokenizer, network, params, pad_token_id if kwargs.get("pad_token_id"):