Skip to content
代码片段 群组 项目
提交 d31df04c 编辑于 作者: comfyanonymous's avatar comfyanonymous
浏览文件

Indentation.

上级 e68763f4
No related branches found
No related tags found
无相关合并请求
......@@ -115,14 +115,14 @@ class Flux(nn.Module):
pe = self.pe_embedder(ids)
for i, block in enumerate(self.double_blocks):
img, txt = block(img=img, txt=txt, vec=vec, pe=pe)
if control is not None: # Controlnet
control_i = control.get("input")
if i < len(control_i):
add = control_i[i]
if add is not None:
img += add
img, txt = block(img=img, txt=txt, vec=vec, pe=pe)
if control is not None: # Controlnet
control_i = control.get("input")
if i < len(control_i):
add = control_i[i]
if add is not None:
img += add
img = torch.cat((txt, img), 1)
......
0% 加载中 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册