nodes/lldacing-ComfyUI_Patches_ll
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: latent_shapes error

Browse Source
This commit is contained in:
刘雪峰
2025-11-04 18:36:27 +08:00
parent 8db65b0b9b
commit ebff62ede8
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/ParaAttentionNode.py
+2 -2
View File
@@ -129,12 +129,12 @@ def fb_cache_patch_dit_exit(img, transformer_options):
return img
def fb_cache_prepare_wrapper(wrapper_executor, noise, latent_image, sampler, sigmas, denoise_mask=None,
callback=None, disable_pbar=False, seed=None):
callback=None, disable_pbar=False, seed=None, **kwargs):
cfg_guider = wrapper_executor.class_obj
try:
out = wrapper_executor(noise, latent_image, sampler, sigmas, denoise_mask=denoise_mask, callback=callback,
disable_pbar=disable_pbar, seed=seed)
disable_pbar=disable_pbar, seed=seed, **kwargs)
finally:
diffusion_model = cfg_guider.model_patcher.model.diffusion_model
if hasattr(diffusion_model, fb_cache_model_temp):
nodes/TeaCacheNode.py
+2 -2
View File
@@ -225,7 +225,7 @@ def tea_cache_patch_dit_exit(img, transformer_options):
return img
def tea_cache_prepare_wrapper(wrapper_executor, noise, latent_image, sampler, sigmas, denoise_mask=None,
callback=None, disable_pbar=False, seed=None):
callback=None, disable_pbar=False, seed=None, **kwargs):
cfg_guider = wrapper_executor.class_obj
# Use cfd_guider.model_options, which is copied from modelPatcher.model_options and will be restored after execution without any unexpected contamination
@@ -234,7 +234,7 @@ def tea_cache_prepare_wrapper(wrapper_executor, noise, latent_image, sampler, si
temp_options['cnt'] = 0
try:
out = wrapper_executor(noise, latent_image, sampler, sigmas, denoise_mask=denoise_mask, callback=callback,
disable_pbar=disable_pbar, seed=seed)
disable_pbar=disable_pbar, seed=seed, **kwargs)
finally:
diffusion_model = cfg_guider.model_patcher.model.diffusion_model
if hasattr(diffusion_model, "flux_tea_cache"):