Spaces:

HarBat
/

ChatBOT

Runtime error

HarBat commited on Jul 25, 2023

Commit

0fa9783

1 Parent(s): e6542f2

Update chain.py

Files changed (1) hide show

chain.py CHANGED Viewed

@@ -65,7 +65,7 @@ def create_question_answering_chain(retriever):
         torch_dtype = torch.bfloat16,
         trust_remote_code = True,
         device_map = "auto",
-        max_length = 1000,
         do_sample = True,
         top_k = 10,
         num_return_sequences = 1,

         torch_dtype = torch.bfloat16,
         trust_remote_code = True,
         device_map = "auto",
+        max_length = 4096,
         do_sample = True,
         top_k = 10,
         num_return_sequences = 1,