File tree Expand file tree Collapse file tree 1 file changed +5
-7
lines changed
examples/research_projects/anytext Expand file tree Collapse file tree 1 file changed +5
-7
lines changed Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments