xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-22 19:34:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update node.py

Browse Source
This commit is contained in:
caojiachen1 2024-09-07 13:20:01 +08:00
parent 0e9010f9f6
commit b300ce3692
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
nodes.py
View File

@ -98,11 +98,11 @@ class DownloadAndLoadCogVideoModel:
os.environ['NEXFORT_FX_FORCE_TRITON_SDPA'] = '1' os.environ['NEXFORT_FX_FORCE_TRITON_SDPA'] = '1'
pipe = compile_pipe( pipe = compile_pipe(
pipe, pipe,
backend="nexfort", backend="nexfort",
options= {"mode": "max-optimize:max-autotune:max-autotune", "memory_format": "channels_last", "options": {"inductor.optimize_linear_epilogue": False, "triton.fuse_attention_allow_fp16_reduction": False}}, options= {"mode": "max-optimize:max-autotune:max-autotune", "memory_format": "channels_last", "options": {"inductor.optimize_linear_epilogue": False, "triton.fuse_attention_allow_fp16_reduction": False}},
ignores=["vae"], ignores=["vae"],
fuse_qkv_projections=True, fuse_qkv_projections=True,
) )
pipeline = { pipeline = {