From 4627b389ff07efea282ec47a58411ab78d920195 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Sat, 1 Apr 2023 09:15:57 +0900 Subject: [PATCH] fix device not specified in merge_lora.py --- library/model_util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/library/model_util.py b/library/model_util.py index b5833066..35b0b6af 100644 --- a/library/model_util.py +++ b/library/model_util.py @@ -812,7 +812,7 @@ def is_safetensors(path): return os.path.splitext(path)[1].lower() == ".safetensors" -def load_checkpoint_with_text_encoder_conversion(ckpt_path, device): +def load_checkpoint_with_text_encoder_conversion(ckpt_path, device="cpu"): # text encoderの格納形式が違うモデルに対応する ('text_model'がない) TEXT_ENCODER_KEY_REPLACEMENTS = [ ("cond_stage_model.transformer.embeddings.", "cond_stage_model.transformer.text_model.embeddings."),