Spaces:

TeamSAS
/

UB_VSA

Running

SakshamLak commited on May 1

Commit

81a46db

verified ·

1 Parent(s): a0505b5

Update buffalo_rag/model/rag.py

Files changed (1) hide show

buffalo_rag/model/rag.py CHANGED Viewed

@@ -73,7 +73,7 @@ class BuffaloRAG:
                         "content": prompt
                     }
                 ],
-                max_tokens=512,
             )
             return completion.choices[0].message.content

                         "content": prompt
                     }
                 ],
+                max_tokens=2048,
             )
             return completion.choices[0].message.content