We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cd4e102 commit 12cc587Copy full SHA for 12cc587
diffsynth_engine/models/flux/flux_dit_fbcache.py
@@ -188,7 +188,7 @@ def from_state_dict(
188
dtype: torch.dtype,
189
in_channel: int = 64,
190
attn_kwargs: Optional[Dict[str, Any]] = None,
191
- fb_cache_relative_l1_threshold: float = 0.05,
+ relative_l1_threshold: float = 0.05,
192
):
193
with no_init_weights():
194
model = torch.nn.utils.skip_init(
@@ -197,7 +197,7 @@ def from_state_dict(
197
dtype=dtype,
198
in_channel=in_channel,
199
attn_kwargs=attn_kwargs,
200
- fb_cache_relative_l1_threshold=fb_cache_relative_l1_threshold,
+ relative_l1_threshold=relative_l1_threshold,
201
)
202
model = model.requires_grad_(False) # for loading gguf
203
model.load_state_dict(state_dict, assign=True)
0 commit comments