manbeast3b
/

fire_stang2

Model card Files Files and versions

manbeast3b commited on Feb 23, 2025

Commit

ef44385

·

verified ·

1 Parent(s): e3a7d1b

Update src/caching.py

Files changed (1) hide show

src/caching.py +1 -1

src/caching.py CHANGED Viewed

@@ -225,7 +225,7 @@ class CachedTransformerBlocks(torch.nn.Module):
 def apply_cache_on_transformer(
     transformer: FluxTransformer2DModel,
     *,
-    residual_diff_threshold=0.06,
 ):
     cached_transformer_blocks = torch.nn.ModuleList(
         [

 def apply_cache_on_transformer(
     transformer: FluxTransformer2DModel,
     *,
+    residual_diff_threshold=0.1,
 ):
     cached_transformer_blocks = torch.nn.ModuleList(
         [