zach9111 commited on
Commit
7821233
·
verified ·
1 Parent(s): b76c0ed

Update src/rag_engine.py

Browse files
Files changed (1) hide show
  1. src/rag_engine.py +2 -2
src/rag_engine.py CHANGED
@@ -19,8 +19,8 @@ clip_model.eval().to("cpu")
19
 
20
  qa_pipeline = pipeline(
21
  "text2text-generation",
22
- model="google/flan-t5-base",
23
- tokenizer="google/flan-t5-base",
24
  device=-1 # CPU
25
  )
26
 
 
19
 
20
  qa_pipeline = pipeline(
21
  "text2text-generation",
22
+ model="google/flan-t5-small",
23
+ tokenizer="google/flan-t5-small",
24
  device=-1 # CPU
25
  )
26