Skip to content

Commit 2cbdf35

Browse files
remove initial_input_channels change for val
1 parent a9f448e commit 2cbdf35

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

examples/cogview4-control/train_control_cogview4.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -804,7 +804,6 @@ def main(args):
804804
cogview4_transformer.patch_embed.proj = new_linear
805805

806806
assert torch.all(cogview4_transformer.patch_embed.proj.weight[:, initial_input_channels:].data == 0)
807-
cogview4_transformer.register_to_config(in_channels=initial_input_channels * 2, out_channels=initial_input_channels)
808807

809808
if args.only_target_transformer_blocks:
810809
cogview4_transformer.patch_embed.proj.requires_grad_(True)

0 commit comments

Comments
 (0)