From 4a70030ccb87e3ea31e9e3416ac8383cdc99d58a Mon Sep 17 00:00:00 2001 From: YunjieYu Date: Tue, 18 Mar 2025 19:24:31 +0800 Subject: [PATCH] fix bug for flux pulid --- nodes.py | 6 ++++-- pyproject.toml | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/nodes.py b/nodes.py index 0b174fe..df190b3 100644 --- a/nodes.py +++ b/nodes.py @@ -118,7 +118,8 @@ def teacache_flux_forward( "vec": vec, "pe": pe, "attn_mask": attn_mask}, - {"original_block": block_wrap}) + {"original_block": block_wrap, + "transformer_options": transformer_options}) txt = out["txt"] img = out["img"] else: @@ -161,7 +162,8 @@ def teacache_flux_forward( "vec": vec, "pe": pe, "attn_mask": attn_mask}, - {"original_block": block_wrap}) + {"original_block": block_wrap, + "transformer_options": transformer_options}) img = out["img"] else: img = block(img, vec=vec, pe=pe, attn_mask=attn_mask) diff --git a/pyproject.toml b/pyproject.toml index 8c20eae..532215c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "teacache" description = "Unofficial implementation of [ali-vilab/TeaCache](https://github.com/ali-vilab/TeaCache) for ComfyUI" -version = "1.4.3" +version = "1.4.4" license = {file = "LICENSE"} [project.urls]