From 2a3deaa06434f44aa506fa20c06aead6a8f152a0 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 9 Oct 2025 22:46:27 +0300 Subject: [PATCH] Check T5 keys before override --- modules/sdnq/loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/sdnq/loader.py b/modules/sdnq/loader.py index a66e767a5..375a336b6 100644 --- a/modules/sdnq/loader.py +++ b/modules/sdnq/loader.py @@ -100,7 +100,7 @@ def load_sdnq_model(model_path: str, model_cls: ModelMixin = None, file_name: st for k in f.keys(): state_dict[k] = f.get_tensor(k) - if model.__class__.__name__ in {"T5EncoderModel", "UMT5EncoderModel"}: + if model.__class__.__name__ in {"T5EncoderModel", "UMT5EncoderModel"} and "encoder.embed_tokens.weight" not in state_dict.keys(): state_dict["encoder.embed_tokens.weight"] = state_dict["shared.weight"] model.load_state_dict(state_dict, assign=True)