Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-04 14:16:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed diffusion forcing

Browse Source
This commit is contained in:
deepbeepmeep 2025-07-30 23:04:43 +02:00
parent b019966b82
commit 6fc95f1db5
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
wgp.py
View File

@ -2631,10 +2631,8 @@ def load_wan_model(model_filename, model_type, base_model_type, model_def, quant
) )
pipe = {"transformer": wan_model.model, "text_encoder" : wan_model.text_encoder.model, "vae": wan_model.vae.model } pipe = {"transformer": wan_model.model, "text_encoder" : wan_model.text_encoder.model, "vae": wan_model.vae.model }
if wan_model.model2 is not None: if hasattr(wan_model,"model2") and wan_model.model2 is not None:
pipe["transformer2"] = wan_model.model2 pipe["transformer2"] = wan_model.model2
# del pipe["transformer"]
# pipe["transformer"] = wan_model.model
if hasattr(wan_model, "clip"): if hasattr(wan_model, "clip"):
pipe["text_encoder_2"] = wan_model.clip.model pipe["text_encoder_2"] = wan_model.clip.model
return wan_model, pipe return wan_model, pipe