phi commit

utils.py

@@ -13,7 +13,7 @@ from pypdf import PdfMerger
 from argparse import ArgumentParser
 
 
-mod = "google/gemma-2b"
+mod = "microsoft/phi-1_5"
 tsk = "text-generation"
 
 def merge_pdfs(pdfs: list):
@@ -132,12 +132,15 @@ def just_chatting(
 try:
     tokenizer = AutoTokenizer.from_pretrained(
         mod,
+        trust_remote_code=True,
     )
 
     tokenizer.pad_token = tokenizer.eos_token
 
     model = AutoModelForCausalLM.from_pretrained(
         mod,
+        trust_remote_code=True,
+        torch_dtype="auto",
     )
 except Exception as e:
     import sys