Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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",
|
|
|
|
|
|
|
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 |
|