VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove print

Browse Source
This commit is contained in:
Kohya S
2023-02-14 19:25:43 +09:00
parent cebee02698
commit 64bffe5238
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
networks/control_net_lora.py
View File

@@ -226,7 +226,7 @@ class ControlLoRANetwork(torch.nn.Module):
sample, res_samples = downsample_block(hidden_states=sample, temb=emb) sample, res_samples = downsample_block(hidden_states=sample, temb=emb)
if ctrl_outs is None: if ctrl_outs is None:
for rs in res_samples: for rs in res_samples:
print("zc", zc_idx, rs.size()) # print("zc", zc_idx, rs.size())
outs.append(self.control_model.zero_convs[zc_idx][0](rs)) # , emb, encoder_hidden_states)) outs.append(self.control_model.zero_convs[zc_idx][0](rs)) # , emb, encoder_hidden_states))
zc_idx += 1 zc_idx += 1
@@ -252,7 +252,6 @@ class ControlLoRANetwork(torch.nn.Module):
apply_ctrl_outs = ctrl_outs[-len(res_samples):] apply_ctrl_outs = ctrl_outs[-len(res_samples):]
ctrl_outs = ctrl_outs[:-len(res_samples)] ctrl_outs = ctrl_outs[:-len(res_samples)]
for j in range(len(res_samples)): for j in range(len(res_samples)):
print(i, j)
res_samples[j] = res_samples[j] + apply_ctrl_outs[j] res_samples[j] = res_samples[j] + apply_ctrl_outs[j]
res_samples = tuple(res_samples) res_samples = tuple(res_samples)