From c4b98d26050227d7b53a54437302be5aa412b70e Mon Sep 17 00:00:00 2001 From: Xiao Date: Sat, 16 Mar 2024 13:48:42 +0000 Subject: [PATCH] Upload XLMRobertaForSequenceClassification --- config.json | 34 ++++++++++++++++++++++++++++++++++ model.safetensors | 3 +++ 2 files changed, 37 insertions(+) create mode 100644 config.json create mode 100644 model.safetensors diff --git a/config.json b/config.json new file mode 100644 index 0000000..a8b22bc --- /dev/null +++ b/config.json @@ -0,0 +1,34 @@ +{ + "_name_or_path": "/share/cf/pycharm/reranker-finetune/bge_rerank/finetune_for_m3/m3_reranker", + "architectures": [ + "XLMRobertaForSequenceClassification" + ], + "attention_probs_dropout_prob": 0.1, + "bos_token_id": 0, + "classifier_dropout": null, + "eos_token_id": 2, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.1, + "hidden_size": 1024, + "id2label": { + "0": "LABEL_0" + }, + "initializer_range": 0.02, + "intermediate_size": 4096, + "label2id": { + "LABEL_0": 0 + }, + "layer_norm_eps": 1e-05, + "max_position_embeddings": 8194, + "model_type": "xlm-roberta", + "num_attention_heads": 16, + "num_hidden_layers": 24, + "output_past": true, + "pad_token_id": 1, + "position_embedding_type": "absolute", + "torch_dtype": "float32", + "transformers_version": "4.38.1", + "type_vocab_size": 1, + "use_cache": true, + "vocab_size": 250002 +} diff --git a/model.safetensors b/model.safetensors new file mode 100644 index 0000000..773a0a7 --- /dev/null +++ b/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d9e3e081faff1eefb84019509b2f5558fd74c1a05a2c7db22f74174fcedb5286 +size 2271071852