Spaces:
Sleeping
Sleeping
Update chatpdf.py
Browse files- chatpdf.py +2 -2
chatpdf.py
CHANGED
@@ -139,7 +139,7 @@ class ChatPDF:
|
|
139 |
self,
|
140 |
similarity_model: SimilarityABC = None,
|
141 |
generate_model_type: str = "auto",
|
142 |
-
generate_model_name_or_path: str = "
|
143 |
lora_model_name_or_path: str = None,
|
144 |
corpus_files: Union[str, List[str]] = None,
|
145 |
save_corpus_emb_dir: str = "corpus_embs/",
|
@@ -566,7 +566,7 @@ if __name__ == "__main__":
|
|
566 |
parser = argparse.ArgumentParser()
|
567 |
parser.add_argument("--sim_model_name", type=str, default="shibing624/text2vec-base-multilingual")
|
568 |
parser.add_argument("--gen_model_type", type=str, default="auto")
|
569 |
-
parser.add_argument("--gen_model_name", type=str, default="
|
570 |
parser.add_argument("--lora_model", type=str, default=None)
|
571 |
parser.add_argument("--rerank_model_name", type=str, default="")
|
572 |
parser.add_argument("--corpus_files", type=str, default="Acuerdo009.pdf")
|
|
|
139 |
self,
|
140 |
similarity_model: SimilarityABC = None,
|
141 |
generate_model_type: str = "auto",
|
142 |
+
generate_model_name_or_path: str = "Qwen/Qwen2-0.5B-Instruct",
|
143 |
lora_model_name_or_path: str = None,
|
144 |
corpus_files: Union[str, List[str]] = None,
|
145 |
save_corpus_emb_dir: str = "corpus_embs/",
|
|
|
566 |
parser = argparse.ArgumentParser()
|
567 |
parser.add_argument("--sim_model_name", type=str, default="shibing624/text2vec-base-multilingual")
|
568 |
parser.add_argument("--gen_model_type", type=str, default="auto")
|
569 |
+
parser.add_argument("--gen_model_name", type=str, default="Qwen/Qwen2-0.5B-Instruct")
|
570 |
parser.add_argument("--lora_model", type=str, default=None)
|
571 |
parser.add_argument("--rerank_model_name", type=str, default="")
|
572 |
parser.add_argument("--corpus_files", type=str, default="Acuerdo009.pdf")
|