Researcher / src /defaults.py
sidphbot's picture
print fix, title model switch
da318a3
# defaults for arxiv
DEFAULTS_HIGH_GPU = {
"max_search": 100,
"num_papers": 20,
"high_gpu": False,
"pdf_dir": "arxiv_data/tarpdfs/",
"txt_dir": "arxiv_data/fulltext/",
"img_dir": "arxiv_data/images/",
"tab_dir": "arxiv_data/tables/",
"dump_dir": "arxiv_dumps/",
"models_dir": "saved_models/",
"title_model_name": "Callidior/bert2bert-base-arxiv-titlegen",
"ex_summ_model_name": "allenai/scibert_scivocab_uncased",
"ledmodel_name": "allenai/led-large-16384-arxiv",
"embedder_name": "paraphrase-MiniLM-L6-v2",
"nlp_name": "en_core_sci_scibert",
"similarity_nlp_name": "en_core_sci_lg",
"kw_model_name": "distilbert-base-nli-mean-tokens",
}
DEFAULTS_CPU_COMPAT = {
"max_search": 100,
"num_papers": 20,
"high_gpu": False,
"pdf_dir": "arxiv_data/tarpdfs/",
"txt_dir": "arxiv_data/fulltext/",
"img_dir": "arxiv_data/images/",
"tab_dir": "arxiv_data/tables/",
"dump_dir": "arxiv_dumps/",
"models_dir": "saved_models/",
"title_model_name": "Callidior/bert2bert-base-arxiv-titlegen",
"ex_summ_model_name": "allenai/scibert_scivocab_uncased",
"ledmodel_name": "ccdv/lsg-bart-base-4096-arxiv",
"embedder_name": "paraphrase-MiniLM-L6-v2",
"nlp_name": "en_core_sci_scibert",
"similarity_nlp_name": "en_core_sci_lg",
"kw_model_name": "distilbert-base-nli-mean-tokens",
}