Yoxas commited on
Commit
57875ae
·
verified ·
1 Parent(s): 6b02087

Update src/pdfchatbot.py

Browse files
Files changed (1) hide show
  1. 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 = 512
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=1024,
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,