add model

Files changed (3) hide show

config.json CHANGED Viewed

@@ -2,7 +2,7 @@
   "architectures": [
     "RagSequenceForGeneration"
   ],
-  "context_max_length": 256,
   "dataset": "wiki_dpr",
   "dataset_split": "train",
   "do_deduplication": true,
@@ -126,10 +126,10 @@
   "label_smoothing": 0.0,
   "max_combined_length": 300,
   "model_type": "rag",
-  "n_docs": 10,
   "output_retrieved": false,
   "passages_path": "test_run/dummy-kb/my_knowledge_dataset",
-  "query_max_length": 64,
   "question_encoder": {
     "_name_or_path": "hfl/chinese-macbert-base",
     "add_cross_attention": false,

   "architectures": [
     "RagSequenceForGeneration"
   ],
+  "context_max_length": 128,
   "dataset": "wiki_dpr",
   "dataset_split": "train",
   "do_deduplication": true,
   "label_smoothing": 0.0,
   "max_combined_length": 300,
   "model_type": "rag",
+  "n_docs": 5,
   "output_retrieved": false,
   "passages_path": "test_run/dummy-kb/my_knowledge_dataset",
+  "query_max_length": 32,
   "question_encoder": {
     "_name_or_path": "hfl/chinese-macbert-base",
     "add_cross_attention": false,

generator_tokenizer/tokenizer.json ADDED Viewed

The diff for this file is too large to render. See raw diff

pytorch_model.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:a388fd66419b7669eb9cb6b27b8687341a510beb47f78606fb8cfef5794ba2d3
+size 874340393