diff --git a/config.json b/config.json index cdf9c3a..a2a88b9 100644 --- a/config.json +++ b/config.json @@ -58,6 +58,7 @@ "output_scores": false, "pad_token_id": 1, "prefix": null, + "projection_dim": 512, "problem_type": null, "pruned_heads": {}, "remove_invalid_values": false, @@ -132,6 +133,7 @@ "pad_token_id": null, "patch_size": 32, "prefix": null, + "projection_dim" : 512, "problem_type": null, "pruned_heads": {}, "remove_invalid_values": false,