WilliamGazeley commited on
Commit
109014c
1 Parent(s): aaac205

Try set LD lib path

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