clementsan commited on
Commit
7ab35ce
1 Parent(s): e384a1c

Remove gradio error for phi2 LLM

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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"}