rag-token-nq / config.json
system's picture
system HF staff
Update config.json
8a39a84
raw
history blame
617 Bytes
{
"bos_token_id": 0,
"decoder_start_token_id": 2,
"doc_sep": " // ",
"eos_token_id": 2,
"is_encoder_decoder": true,
"max_combined_length": 300,
"model_type": "rag",
"n_docs": 5,
"pad_token_id": 1,
"pretrained_generator_tokenizer_name_or_path": "facebook/bart-large",
"pretrained_question_encoder_tokenizer_name_or_path": "facebook/dpr-question_encoder-single-nq-base",
"retrieval_batch_size": 8,
"retrieval_vector_size": 768,
"retriever_type": "hf_retriever",
"dataset": "wiki_dpr",
"dataset_split": "train",
"index_name": "embeddings",
"title_sep": " / ",
"vocab_size": 50265
}