manbeast3b
/

fire_stang4

Model card Files Files and versions Community

manbeast3b commited on Feb 23

Commit

848125f

·

verified ·

1 Parent(s): 471b188

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.1,
 ):
     cached_transformer_blocks = torch.nn.ModuleList(
         [

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