initial config

config.json

@@ -1 +1 @@
-{"embed_dim": 64, "n_heads": 2, "n_layers": 2, "ff_dim": 128}
+{"embed_dim": 128, "n_heads": 4, "n_layers": 4, "ff_dim": 256}