From b300ce3692c27685a1449ce51fdf8565df2b0f7d Mon Sep 17 00:00:00 2001 From: caojiachen1 Date: Sat, 7 Sep 2024 13:20:01 +0800 Subject: [PATCH] Update node.py --- nodes.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/nodes.py b/nodes.py index dfce948..326a9e7 100644 --- a/nodes.py +++ b/nodes.py @@ -98,11 +98,11 @@ class DownloadAndLoadCogVideoModel: os.environ['NEXFORT_FX_FORCE_TRITON_SDPA'] = '1' pipe = compile_pipe( - pipe, - 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}}, - ignores=["vae"], - fuse_qkv_projections=True, + pipe, + 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}}, + ignores=["vae"], + fuse_qkv_projections=True, ) pipeline = {