File size: 1,062 Bytes
f25ee03 6789388 f25ee03 b93cab2 f25ee03 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
{
"model_type": "hourglass_transformer",
"auto_map": {
"AutoConfig": "hourglass_transformer.HourglassTransformerConfig",
"AutoModel": "hourglass_transformer.HourglassTransformerForMaskedLM"
},
"activation_function": "gelu",
"architectures": [
"HourglassTransformerForMaskedLM"
],
"attn_resampling": false,
"bias": false,
"depth": [
4,
[
4,
4,
4
],
4
],
"dim": 768,
"dim_head": 64,
"heads": 8,
"inference": true,
"metadata_dim": 3072,
"model_type": "hourglass_transformer",
"norm_out": true,
"predict_expression_mode": false,
"predict_seq": true,
"predict_taxonomy": false,
"predict_tracks": false,
"rotary_emb_dim": 32,
"seq_vocab_size": 11,
"shorten_factor": [
8,
8
],
"sliding_window": [
512,
512,
-1
],
"taxonomy_vocab_size": 2604,
"torch_dtype": "float32",
"track_activation_fn": null,
"track_output_dim": 4,
"transformers_version": "4.44.2",
"updown_sample_type": "linear",
"use_metadata": false,
"use_taxonomy": false
}
|