test pr

0_Transformer/config.json

@@ -1,31 +0,0 @@
-{
-  "_name_or_path": "answerdotai/answerai-colbert-small-v1",
-  "architectures": [
-    "BertModel"
-  ],
-  "attention_probs_dropout_prob": 0.1,
-  "classifier_dropout": null,
-  "hidden_act": "gelu",
-  "hidden_dropout_prob": 0.1,
-  "hidden_size": 384,
-  "id2label": {
-    "0": "LABEL_0"
-  },
-  "initializer_range": 0.02,
-  "intermediate_size": 1536,
-  "label2id": {
-    "LABEL_0": 0
-  },
-  "layer_norm_eps": 1e-12,
-  "max_position_embeddings": 512,
-  "model_type": "bert",
-  "num_attention_heads": 12,
-  "num_hidden_layers": 12,
-  "pad_token_id": 0,
-  "position_embedding_type": "absolute",
-  "torch_dtype": "float32",
-  "transformers_version": "4.45.0.dev0",
-  "type_vocab_size": 2,
-  "use_cache": true,
-  "vocab_size": 30522
-}

0_Transformer/model.safetensors

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:ac67445c20ea550edb254061e4e9402095f99bcba9c0b9a952e5f005abde9f1a
-size 132870584

0_Transformer/sentence_bert_config.json

@@ -1,7 +0,0 @@
-{
-  "max_seq_length": 512,
-  "do_lower_case": false,
-  "model_args": {
-    "add_pooling_layer": false
-  }
-}

1_Dense/config.json

@@ -1 +0,0 @@
-{"in_features": 384, "out_features": 96, "bias": false, "activation_function": "torch.nn.modules.linear.Identity"}

1_Dense/model.safetensors

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:47d2690e7c1612c84b7b941e69a883e99190b9b44ebb0be00152f68966c6eb11
-size 147544

README.md

@@ -54,7 +54,7 @@ RAG = RAGPretrainedModel.from_pretrained("answerdotai/answerai-colbert-small-v1"
 
 docs = ['Hayao Miyazaki is a Japanese director, born on [...]', 'Walt Disney is an American author, director and [...]', ...]
 
-RAG.index(docs, index_name="ghibli")
+RAG.index(documents, index_name="ghibli")
 
 query = 'Who directed spirited away?'
 results = RAG.search(query)

config_sentence_transformers.json

@@ -1,49 +0,0 @@
-{
-  "__version__": {
-    "sentence_transformers": "3.0.1",
-    "transformers": "4.45.0.dev0",
-    "pytorch": "2.4.0+cu121"
-  },
-  "prompts": {},
-  "default_prompt_name": null,
-  "similarity_fn_name": null,
-  "query_prefix": "[unused0]",
-  "document_prefix": "[unused1]",
-  "query_length": 32,
-  "document_length": 300,
-  "attend_to_expansion_tokens": false,
-  "skiplist_words": [
-    "!",
-    "\"",
-    "#",
-    "$",
-    "%",
-    "&",
-    "'",
-    "(",
-    ")",
-    "*",
-    "+",
-    ",",
-    "-",
-    ".",
-    "/",
-    ":",
-    ";",
-    "<",
-    "=",
-    ">",
-    "?",
-    "@",
-    "[",
-    "\\",
-    "]",
-    "^",
-    "_",
-    "`",
-    "{",
-    "|",
-    "}",
-    "~"
-  ]
-}

modules.json

@@ -1,14 +0,0 @@
-[
-  {
-    "idx": 0,
-    "name": "0",
-    "path": "0_Transformer",
-    "type": "sentence_transformers.models.Transformer"
-  },
-  {
-    "idx": 1,
-    "name": "1",
-    "path": "1_Dense",
-    "type": "pylate.models.Dense"
-  }
-]