clementsan
commited on
Commit
•
7ab35ce
1
Parent(s):
e384a1c
Remove gradio error for phi2 LLM
Browse files
app.py
CHANGED
@@ -120,7 +120,7 @@ def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, pr
|
|
120 |
top_k = top_k,
|
121 |
)
|
122 |
elif llm_model == "microsoft/phi-2":
|
123 |
-
raise gr.Error("phi-2 model requires 'trust_remote_code=True', currently not supported by langchain HuggingFaceHub...")
|
124 |
llm = HuggingFaceEndpoint(
|
125 |
repo_id=llm_model,
|
126 |
# model_kwargs={"temperature": temperature, "max_new_tokens": max_tokens, "top_k": top_k, "trust_remote_code": True, "torch_dtype": "auto"}
|
|
|
120 |
top_k = top_k,
|
121 |
)
|
122 |
elif llm_model == "microsoft/phi-2":
|
123 |
+
# raise gr.Error("phi-2 model requires 'trust_remote_code=True', currently not supported by langchain HuggingFaceHub...")
|
124 |
llm = HuggingFaceEndpoint(
|
125 |
repo_id=llm_model,
|
126 |
# model_kwargs={"temperature": temperature, "max_new_tokens": max_tokens, "top_k": top_k, "trust_remote_code": True, "torch_dtype": "auto"}
|