add cond/uncond, update config

This commit is contained in:
Kohya S
2023-08-17 16:25:23 +09:00
parent 784a90c3a6
commit 5fa473d5f3
2 changed files with 26 additions and 11 deletions

View File

@@ -1571,7 +1571,7 @@ def main(args):
control_net.apply_to()
control_net.load_state_dict(state_dict)
control_net.to(dtype).to(device)
control_net.set_batch_cond_uncond_enabled(True)
control_net.set_batch_cond_only(False, False)
control_nets.append(control_net)
if args.opt_channels_last: