We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a9f448e commit 2cbdf35Copy full SHA for 2cbdf35
examples/cogview4-control/train_control_cogview4.py
@@ -804,7 +804,6 @@ def main(args):
804
cogview4_transformer.patch_embed.proj = new_linear
805
806
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)
808
809
if args.only_target_transformer_blocks:
810
cogview4_transformer.patch_embed.proj.requires_grad_(True)
0 commit comments