{ "architecture": { "embedding_size": 768, "hidden_size": 768, "intermediate_size": 3072, "num_hidden_layers": 12, "num_attention_heads": 12, "attention_head_size": 64, "attention_probs_dropout_prob": 0.1, "hidden_dropout_prob": 0.1, "hidden_act": "gelu", "initializer_range": 0.02, "layer_norm_eps": 1e-07, "max_position_embeddings": 0, "max_relative_positions": 512, "position_buckets": 0, "norm_rel_ebd": "none", "pos_att_type": "c2p", "position_biased_input": false, "relative_attention": true, "share_att_key": false, "type_vocab_size": 0, "vocab_size": 211, "chemistry_bond_features": true, "chemistry_stereo_features": true, "chemistry_feature_version": 1 }, "source_checkpoint_sha256": "15da88b4c3e079b127142cc7bd0eb61819fd94d3aba3344ca919f659876c023f", "representation": "final_layer_cls", "model_variant": "A2", "training_stage": "step2", "chemistry_feature_version": 1 }