WilliamGazeley
commited on
Commit
·
558d9e8
1
Parent(s):
eaab710
Remove unused arg
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ sys_msg = "You are a super intelligent automated financial advisor created by IR
|
|
| 8 |
@st.cache_resource(show_spinner=False)
|
| 9 |
def init_llm():
|
| 10 |
huggingface_hub.login(token=os.getenv("HF_TOKEN"))
|
| 11 |
-
llm = LLM(model="InvestmentResearchAI/LLM-ADE-dev"
|
| 12 |
tok = llm.get_tokenizer()
|
| 13 |
tok.eos_token = '<|im_end|>' # Override to use turns
|
| 14 |
return llm
|
|
|
|
| 8 |
@st.cache_resource(show_spinner=False)
|
| 9 |
def init_llm():
|
| 10 |
huggingface_hub.login(token=os.getenv("HF_TOKEN"))
|
| 11 |
+
llm = LLM(model="InvestmentResearchAI/LLM-ADE-dev")
|
| 12 |
tok = llm.get_tokenizer()
|
| 13 |
tok.eos_token = '<|im_end|>' # Override to use turns
|
| 14 |
return llm
|