nodes/welltop-cn-ComfyUI-TeaCache
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix bug for flux pulid

Browse Source
This commit is contained in:
YunjieYu
2025-03-18 19:24:31 +08:00
parent e28a937aa2
commit 4a70030ccb
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+4 -2
View File
@@ -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)
pyproject.toml
+1 -1
View File
@@ -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]