diff --git a/config.json b/config.json index fb7e397..f8ab627 100644 --- a/config.json +++ b/config.json @@ -23,7 +23,7 @@ "do_sample": false, "dropout": 0.1, "early_stopping": false, - "encoder_hidden_size": 768, + "cross_attention_hidden_size": 768, "encoder_no_repeat_ngram_size": 0, "eos_token_id": 2, "finetuning_task": null, @@ -88,7 +88,7 @@ "diversity_penalty": 0.0, "do_sample": false, "early_stopping": false, - "encoder_hidden_size": null, + "cross_attention_hidden_size": null, "encoder_no_repeat_ngram_size": 0, "eos_token_id": null, "finetuning_task": null,