nodes/brianfitzgerald-style_aligned_comfy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Bugfix: register shared norm with the copied model

Browse Source
This commit is contained in:
brianf
2023-12-13 22:29:38 +00:00
parent 28804b7116
commit be8a2deeec
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+2 -2
View File
@@ -214,13 +214,13 @@ class StyleAlignedReferenceSampler:
x0_output = {} x0_output = {}
callback = latent_preview.prepare_callback( callback = latent_preview.prepare_callback(
model, sigmas.shape[-1] - 1, x0_output m, sigmas.shape[-1] - 1, x0_output
) )
# Register shared norms # Register shared norms
share_group_norm = share_norm in ["group", "both"] share_group_norm = share_norm in ["group", "both"]
share_layer_norm = share_norm in ["layer", "both"] share_layer_norm = share_norm in ["layer", "both"]
register_shared_norm(model, share_group_norm, share_layer_norm) register_shared_norm(m, share_group_norm, share_layer_norm)
# Patch cross attn # Patch cross attn
m.set_model_attn1_patch(SharedAttentionProcessor(self.args, scale)) m.set_model_attn1_patch(SharedAttentionProcessor(self.args, scale))