Update src/pdfchatbot.py
Browse files- src/pdfchatbot.py +2 -2
src/pdfchatbot.py
CHANGED
@@ -33,7 +33,7 @@ class PDFChatBot:
|
|
33 |
self.model = None
|
34 |
self.pipeline = None
|
35 |
self.chain = None
|
36 |
-
self.chunk_size =
|
37 |
self.overlap_percentage = 50
|
38 |
self.max_chunks_in_context = 2
|
39 |
self.current_context = None
|
@@ -102,7 +102,7 @@ class PDFChatBot:
|
|
102 |
temp = 0.1
|
103 |
outputs = self.pipe(
|
104 |
prompt,
|
105 |
-
max_new_tokens=
|
106 |
do_sample=True,
|
107 |
temperature=temp,
|
108 |
top_p=0.9,
|
|
|
33 |
self.model = None
|
34 |
self.pipeline = None
|
35 |
self.chain = None
|
36 |
+
self.chunk_size = 32768
|
37 |
self.overlap_percentage = 50
|
38 |
self.max_chunks_in_context = 2
|
39 |
self.current_context = None
|
|
|
102 |
temp = 0.1
|
103 |
outputs = self.pipe(
|
104 |
prompt,
|
105 |
+
max_new_tokens=262144,
|
106 |
do_sample=True,
|
107 |
temperature=temp,
|
108 |
top_p=0.9,
|