Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Check T5 keys before override

Browse Source
This commit is contained in:
Disty0
2025-10-09 22:46:27 +03:00
parent 6995d8c3c6
commit 2a3deaa064
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sdnq/loader.py
+1 -1
View File
@@ -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)