{
  "architectures": [
    "MoleBERTTokenizer"
  ],
  "atom_vocab_size": 1024,
  "gnn_encoder_dropout": 0.1,
  "gnn_encoder_embedding_dim": 300,
  "gnn_encoder_jk": "last",
  "gnn_encoder_layer_hidden_dim": 600,
  "gnn_encoder_num_layers": 5,
  "graph_reconstruct_dropout": 0.1,
  "graph_reconstruct_hidden_dim": 600,
  "re_build_edge": true,
  "torch_dtype": "float32",
  "transformers_version": "4.46.3",
  "vq_commitment_cost": 0.25
}