wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-18 07:35:05 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Reuse wan text encoder

Browse Source
This commit is contained in:
qqingzheng 2026-03-12 22:36:11 +08:00
parent a5c328871d
commit 9b6e0b0677
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/sd.py
View File

@ -1338,8 +1338,9 @@ def load_text_encoder_state_dicts(state_dicts=[], embedding_directory=None, clip
clip_target.tokenizer = comfy.text_encoders.wan.WanT5Tokenizer
tokenizer_data["spiece_model"] = clip_data[0].get("spiece_model", None)
elif clip_type == CLIPType.HELIOS:
clip_target.clip = comfy.text_encoders.helios.te(**t5xxl_detect(clip_data))
clip_target.tokenizer = comfy.text_encoders.helios.HeliosT5Tokenizer
# Helios reuses the WAN UMT5-XXL text encoder stack.
clip_target.clip = comfy.text_encoders.wan.te(**t5xxl_detect(clip_data))
clip_target.tokenizer = comfy.text_encoders.wan.WanT5Tokenizer
tokenizer_data["spiece_model"] = clip_data[0].get("spiece_model", None)
elif clip_type == CLIPType.HIDREAM:
clip_target.clip = comfy.text_encoders.hidream.hidream_clip(**t5xxl_detect(clip_data),