Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update src/rag/pipeline.py
Browse files- src/rag/pipeline.py +1 -1
src/rag/pipeline.py
CHANGED
@@ -87,7 +87,7 @@ class RAGPipeline:
|
|
87 |
model_name_or_path="gpt-4",
|
88 |
default_prompt_template=rag_prompt,
|
89 |
api_key=OPENAI_API_KEY,
|
90 |
-
max_length=
|
91 |
model_kwargs={"temperature": 0.2, "max_tokens": 4096},
|
92 |
)
|
93 |
return prompt_node
|
|
|
87 |
model_name_or_path="gpt-4",
|
88 |
default_prompt_template=rag_prompt,
|
89 |
api_key=OPENAI_API_KEY,
|
90 |
+
max_length=4000,
|
91 |
model_kwargs={"temperature": 0.2, "max_tokens": 4096},
|
92 |
)
|
93 |
return prompt_node
|