{ "architectures": [ "NicheformerForMaskedLM" ], "assay": true, "batch_first": true, "cls_classes": 164, "context_length": 1500, "dim_feedforward": 1024, "dim_model": 512, "dropout": 0.0, "learnable_pe": true, "masking_p": 0.15, "modality": true, "model_type": "nicheformer", "n_tokens": 20340, "nheads": 16, "nlayers": 12, "specie": true, "supervised_task": null, "torch_dtype": "float32", "transformers_version": "4.49.0", "auto_map": { "AutoConfig": "configuration_nicheformer.NicheformerConfig", "AutoModelForMaskedLM": "modeling_nicheformer.NicheformerForMaskedLM", "AutoModel": "modeling_nicheformer.NicheformerModel", "AutoTokenizer": ["tokenization_nicheformer.NicheformerTokenizer", "tokenization_nicheformer.NicheformerTokenizer"] } }