Fiwing the config.json of the model

1_MultiHeadGeneralizedPooling/config.json

@@ -3,4 +3,5 @@
     "token_dim": 768,
     "num_heads": 8,
     "initialize": "random"
+    "pooling_type": 0
 }