Affinity_PPI / prot_t5_onnx /config.json
Delower's picture
Upload folder using huggingface_hub
7ce3c91 verified
Raw
History Blame Contribute Delete
1.8 kB
{
"architectures": ["T5EncoderModel"],
"model_type": "t5",
"framework": "onnx",
"onnx": {
"opset_version": 14,
"providers": [
"CoreMLExecutionProvider",
"CUDAExecutionProvider",
"CPUExecutionProvider"
],
"external_data": true,
"optimization_level": "all",
"fp16": true
},
"d_ff": 2816,
"d_kv": 64,
"d_model": 1024,
"decoder_start_token_id": 0,
"dropout_rate": 0.1,
"eos_token_id": 1,
"feed_forward_proj": "gated-gelu",
"initializer_factor": 1.0,
"is_encoder_decoder": false,
"layer_norm_epsilon": 1e-06,
"num_decoder_layers": 24,
"num_heads": 16,
"num_layers": 24,
"pad_token_id": 0,
"relative_attention_num_buckets": 32,
"tie_word_embeddings": false,
"torch_dtype": "float16",
"transformers_version": "4.21.0",
"use_cache": false,
"vocab_size": 32128,
"task_specific_params": {
"protein_embedding": {
"early_stopping": true,
"length_penalty": 2.0,
"max_length": 1024,
"min_length": 1,
"no_repeat_ngram_size": 3,
"num_beams": 4,
"prefix": ""
}
},
"protein_config": {
"amino_acid_vocab": ["A", "C", "D", "E", "F", "G", "H", "I", "K", "L", "M", "N", "P", "Q", "R", "S", "T", "V", "W", "Y", "X"],
"rare_amino_acids": ["U", "Z", "O", "B"],
"replacement_token": "X",
"sequence_separator": " ",
"preprocessing": {
"rare_aa_replacement": true,
"space_separation": true,
"uppercase_only": true
}
},
"conversion_info": {
"source_model": "Rostlab/prot_t5_xl_half_uniref50-enc",
"conversion_date": "2025-07-27",
"conversion_tool": "torch.onnx.export",
"validation_metrics": {
"avg_cosine_similarity": 1.000244,
"max_absolute_difference": 0.001953,
"test_cases_passed": 4
}
}
}