Skip to content

GitLab

"doc/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "025692aa4d3c36974cf8af19ef6bc83ae042bd32"
Commit 37a86e46 authored by comfyanonymous's avatar comfyanonymous
Browse files

Remove duplicate text_projection key from some saved models.

parent 8daedc5b
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
comfy/diffusers_convert.py
View file @ 37a86e46
...@@ -240,9 +240,9 @@ def convert_text_enc_state_dict_v20(text_enc_dict, prefix=""): ...@@ -240,9 +240,9 @@ def convert_text_enc_state_dict_v20(text_enc_dict, prefix=""):
text_proj = "transformer.text_projection.weight" text_proj = "transformer.text_projection.weight"
if k.endswith(text_proj): if k.endswith(text_proj):
new_state_dict[k.replace(text_proj, "text_projection")] = v.transpose(0, 1).contiguous() new_state_dict[k.replace(text_proj, "text_projection")] = v.transpose(0, 1).contiguous()
else:
relabelled_key = textenc_pattern.sub(lambda m: protected[re.escape(m.group(0))], k) relabelled_key = textenc_pattern.sub(lambda m: protected[re.escape(m.group(0))], k)
new_state_dict[relabelled_key] = v new_state_dict[relabelled_key] = v
for k_pre, tensors in capture_qkv_weight.items(): for k_pre, tensors in capture_qkv_weight.items():
if None in tensors: if None in tensors:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment