ccm commited on
Commit
0d94ca2
·
verified ·
1 Parent(s): f35b338

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -68,7 +68,10 @@ publication_vectorstore = langchain_community.vectorstores.FAISS.load_local(
68
 
69
  # Create the callable LLM
70
  llm = transformers.pipeline(
71
- task="text-generation", model="Qwen/Qwen2.5-7B-Instruct-AWQ", device="cuda"
 
 
 
72
  )
73
 
74
 
 
68
 
69
  # Create the callable LLM
70
  llm = transformers.pipeline(
71
+ task="text-generation",
72
+ model="Qwen/Qwen2.5-7B-Instruct-AWQ",
73
+ device="cuda",
74
+ streamer=transformers.TextStreamer(transformers.AutoTokenizer.from_pretrained("Qwen/Qwen2.5-7B-Instruct-AWQ"))
75
  )
76
 
77