Spaces:
Sleeping
Sleeping
WilliamGazeley
commited on
Commit
·
109014c
1
Parent(s):
aaac205
Try set LD lib path
Browse files- app.py +3 -0
- requirements.txt +0 -1
app.py
CHANGED
@@ -1,6 +1,9 @@
|
|
|
|
1 |
import streamlit as st
|
2 |
from vllm import LLM, SamplingParams
|
3 |
|
|
|
|
|
4 |
llm = LLM(model="InvestmentResearchAI/LLM-ADE-small-v0.1.0")
|
5 |
tok = llm.get_tokenizer()
|
6 |
tok.eos_token = '<|eot_id|>' # Override to use turns
|
|
|
1 |
+
import os
|
2 |
import streamlit as st
|
3 |
from vllm import LLM, SamplingParams
|
4 |
|
5 |
+
os.environ['LD_LIBRARY_PATH'] = '/usr/lib/x86_64-linux-gnu'
|
6 |
+
|
7 |
llm = LLM(model="InvestmentResearchAI/LLM-ADE-small-v0.1.0")
|
8 |
tok = llm.get_tokenizer()
|
9 |
tok.eos_token = '<|eot_id|>' # Override to use turns
|
requirements.txt
CHANGED
@@ -2,5 +2,4 @@ streamlit
|
|
2 |
transformers
|
3 |
torch
|
4 |
vllm
|
5 |
-
flash-attn --no-build-isolation
|
6 |
|
|
|
2 |
transformers
|
3 |
torch
|
4 |
vllm
|
|
|
5 |
|