Phi2-PDF-chat

Sleeping

dinhquangson commited on Feb 8

Commit

13bddc0

•

1 Parent(s): 0c7d70e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -112,7 +112,7 @@ def get_conversation_chain(vectorstore):
     """
     llm = HuggingFaceHub(
-        repo_id="mlabonne/phixtral-2x2_8",
         model_kwargs={"temperature": 0.5, "max_new_tokens": 1024, "max_length": 1048, "top_k": 3, "trust_remote_code": True, "torch_dtype": "auto"},
     )

     """
     llm = HuggingFaceHub(
+        repo_id="TheBloke/phi-2-GGUF",
         model_kwargs={"temperature": 0.5, "max_new_tokens": 1024, "max_length": 1048, "top_k": 3, "trust_remote_code": True, "torch_dtype": "auto"},
     )