zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-24 18:10:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Rewrite causual forcing using custom sampler with KSampler node.

Browse Source
This commit is contained in:
Talmaj Marinc
2026-03-24 13:23:06 +01:00
parent 92f937daa6
commit d1998480b0
6 changed files with 170 additions and 182 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
comfy/supported_models.py
View File

@@ -1167,6 +1167,15 @@ class WAN21_T2V(supported_models_base.BASE):
t5_detect = comfy.text_encoders.sd3_clip.t5_xxl_detect(state_dict, "{}umt5xxl.transformer.".format(pref))
return supported_models_base.ClipTarget(comfy.text_encoders.wan.WanT5Tokenizer, comfy.text_encoders.wan.te(**t5_detect))
class WAN21_CausalAR_T2V(WAN21_T2V):
sampling_settings = {
"shift": 5.0,
}
def get_model(self, state_dict, prefix="", device=None):
return model_base.WAN21_CausalAR(self, device=device)
class WAN21_I2V(WAN21_T2V):
unet_config = {
"image_model": "wan2.1",