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

Fix FLUX module imports for ControlNetFlux.

Browse Source
This commit is contained in:
Ryan Dick
2024-10-02 14:32:32 +00:00
parent 2821ba8967
commit 58eba8bdbd
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/backend/flux/controlnet/controlnet_flux.py
View File

@@ -8,7 +8,7 @@ import torch
from einops import rearrange
from torch import Tensor, nn
from .modules.layers import DoubleStreamBlock, EmbedND, MLPEmbedder, timestep_embedding
from invokeai.backend.flux.modules.layers import DoubleStreamBlock, EmbedND, MLPEmbedder, timestep_embedding
@dataclass
@@ -202,7 +202,6 @@ class ControlNetFlux(nn.Module):
return custom_forward
ckpt_kwargs: Dict[str, Any] = {"use_reentrant": False} if is_torch_version(">=", "1.11.0") else {}
encoder_hidden_states, hidden_states = torch.utils.checkpoint.checkpoint(
create_custom_forward(block),
img,