| { | |
| "_name_or_path": "/fsx-checkpoints/jxm/cde/2024-09-18-supervised-final-bge--epoch-4/checkpoint-1820", | |
| "architecture": "transductive", | |
| "architectures": [ | |
| "DatasetTransformer" | |
| ], | |
| "attn_implementation": null, | |
| "auto_map": { | |
| "AutoConfig": "misc.ContextualModelConfig", | |
| "AutoModel": "model.DatasetTransformer" | |
| }, | |
| "biencoder_pooling_strategy": "mean", | |
| "cache_dir": null, | |
| "config_name": null, | |
| "disable_dropout": true, | |
| "disable_transductive_rotary_embedding": true, | |
| "embedder": "nomic-ai/nomic-bert-2048", | |
| "embedder_rerank": "sentence-transformers/gtr-t5-base", | |
| "embedding_output_dim": null, | |
| "limit_layers": null, | |
| "limit_layers_first_stage": null, | |
| "logit_scale": 50.0, | |
| "max_seq_length": 512, | |
| "model_revision": "main", | |
| "tokenizer_name": null, | |
| "torch_dtype": "float32", | |
| "transductive_corpus_size": 512, | |
| "transductive_sequence_dropout_prob": 0.0, | |
| "transductive_tie_token_embeddings": false, | |
| "transductive_tokens_per_document": 1, | |
| "transformers_version": "4.44.2" | |
| } | |