github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-03 17:35:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add unit test to test the full flow of loading an InstantX ControlNet from a state dict.

Browse Source
This commit is contained in:
Ryan Dick
2024-10-07 17:48:29 +00:00
parent 745b6dbd5d
commit 80bc4ebee3
2 changed files with 25 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/flux/controlnet/diffusers_controlnet_flux.py
View File

@@ -89,11 +89,11 @@ class DiffusersControlNetFlux(torch.nn.Module):
# The following modules are specific to the ControlNet model.
# -----------------------------------------------------------
self.controlnet_blocks = nn.ModuleList([])
for _ in range(len(self.transformer_blocks)):
for _ in range(len(self.double_blocks)):
self.controlnet_blocks.append(zero_module(nn.Linear(self.hidden_size, self.hidden_size)))
self.controlnet_single_blocks = nn.ModuleList([])
for _ in range(len(self.single_transformer_blocks)):
for _ in range(len(self.single_blocks)):
self.controlnet_single_blocks.append(zero_module(nn.Linear(self.hidden_size, self.hidden_size)))
self.is_union = False