diff --git a/config.json b/config.json index 7548707..a059af4 100644 --- a/config.json +++ b/config.json @@ -130,6 +130,7 @@ "num_attention_heads": 12, "num_beam_groups": 1, "num_beams": 1, + "num_channels": 3, "num_hidden_layers": 12, "num_return_sequences": 1, "output_attentions": false,