From 64bffe5238161482b4566338d642e7e8bfbbd572 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Tue, 14 Feb 2023 19:25:43 +0900 Subject: [PATCH] remove print --- networks/control_net_lora.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/networks/control_net_lora.py b/networks/control_net_lora.py index 6a98260e..b330baee 100644 --- a/networks/control_net_lora.py +++ b/networks/control_net_lora.py @@ -226,7 +226,7 @@ class ControlLoRANetwork(torch.nn.Module): sample, res_samples = downsample_block(hidden_states=sample, temb=emb) if ctrl_outs is None: 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)) zc_idx += 1 @@ -252,7 +252,6 @@ class ControlLoRANetwork(torch.nn.Module): apply_ctrl_outs = ctrl_outs[-len(res_samples):] ctrl_outs = ctrl_outs[:-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 = tuple(res_samples)