ZoniaChatbot commited on
Commit
3a3f034
·
verified ·
1 Parent(s): e08b11d

Update chatpdf.py

Browse files
Files changed (1) hide show
  1. 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 = "LenguajeNaturalAI/leniachat-qwen2-1.5B-v0",
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="LenguajeNaturalAI/leniachat-qwen2-1.5B-v0")
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")