Commit f04b603c authored by Hyunsung Lee's avatar Hyunsung Lee Committed by muyangli
Browse files

fix to escape cache reloading

parent a23d2b69
......@@ -9,6 +9,7 @@ from ...caching import utils
def apply_cache_on_transformer(transformer: FluxTransformer2DModel, *, residual_diff_threshold=0.12):
if getattr(transformer, "_is_cached", False):
transformer.cached_transformer_blocks[0].update_threshold(residual_diff_threshold)
return transformer
cached_transformer_blocks = nn.ModuleList(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment