Skip to content

Commit 9c43a65

Browse files
committed
Complete control net conditioning embedding in AnyTextControlNetModel
1 parent 971d6ad commit 9c43a65

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

examples/research_projects/anytext/anytext_controlnet.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -250,13 +250,11 @@ def __init__(
250250
)
251251

252252
# control net conditioning embedding
253-
# TODO: what happens ControlNetModel's self.controlnet_cond_embedding's memory occupation?
254-
# self.controlnet_cond_embedding = AnyTextControlNetConditioningEmbedding(
255-
# conditioning_embedding_channels=block_out_channels[0],
256-
# glyph_channels=conditioning_channels,
257-
# position_channels=conditioning_channels,
258-
# )
259-
self.controlnet_cond_embedding = None
253+
self.controlnet_cond_embedding = AnyTextControlNetConditioningEmbedding(
254+
conditioning_embedding_channels=block_out_channels[0],
255+
glyph_channels=conditioning_channels,
256+
position_channels=conditioning_channels,
257+
)
260258

261259
def forward(
262260
self,

0 commit comments

Comments
 (0)