{ "_name_or_path": "albert/albert-base-v2", "architectures": [ "AlbertForSequenceClassification" ], "attention_probs_dropout_prob": 0, "bos_token_id": 2, "classifier_dropout_prob": 0.1, "down_scale_factor": 1, "embedding_size": 128, "eos_token_id": 3, "gap_size": 0, "hidden_act": "gelu_new", "hidden_dropout_prob": 0, "hidden_size": 768, "id2label": { "0": "Noah Kahan", "1": "21 Savage", "2": "Travis Scott", "3": "Taylor Swift", "4": "Jack Harlow", "5": "Megan Thee Stallion", "6": "Doja Cat", "7": "The Grateful Dead", "8": "SZA", "9": "Drake", "10": "Jelly Roll", "11": "Morgan Wallen", "12": "Chris Stapleton", "13": "Olivia Rodrigo", "14": "Zach Bryan", "15": "Luke Combs", "16": "Dua Lipa", "17": "Nicki Minaj", "18": "The Smile", "19": "The Weeknd" }, "initializer_range": 0.02, "inner_group_num": 1, "intermediate_size": 3072, "label2id": { "21 Savage": 1, "Chris Stapleton": 12, "Doja Cat": 6, "Drake": 9, "Dua Lipa": 16, "Jack Harlow": 4, "Jelly Roll": 10, "Luke Combs": 15, "Megan Thee Stallion": 5, "Morgan Wallen": 11, "Nicki Minaj": 17, "Noah Kahan": 0, "Olivia Rodrigo": 13, "SZA": 8, "Taylor Swift": 3, "The Grateful Dead": 7, "The Smile": 18, "The Weeknd": 19, "Travis Scott": 2, "Zach Bryan": 14 }, "layer_norm_eps": 1e-12, "max_position_embeddings": 512, "model_type": "albert", "net_structure_type": 0, "num_attention_heads": 12, "num_hidden_groups": 1, "num_hidden_layers": 12, "num_memory_blocks": 0, "pad_token_id": 0, "position_embedding_type": "absolute", "problem_type": "single_label_classification", "torch_dtype": "float32", "transformers_version": "4.38.2", "type_vocab_size": 2, "vocab_size": 30000 }