(self)
| 144 | return parsed_files |
| 145 | |
| 146 | def load_query_engine(self): |
| 147 | print('Loading nodes...') |
| 148 | self.nodes = self.load_nodes() |
| 149 | if self.vl_ret and 'vidore' in self.embed_model_name: |
| 150 | self.embedding_img = [torch.tensor(node.embedding).view(-1,128).bfloat16() for node in self.nodes] |
| 151 | self.embedding_img = [tensor.to(self.vector_embed_model.embed_model.device) for tensor in self.embedding_img] |
| 152 | else: |
| 153 | retriever = self.load_retriever_embed(self.nodes) |
| 154 | node_postprocessors = self.load_node_postprocessors() |
| 155 | query_engine = RetrieverQueryEngine( |
| 156 | retriever=retriever, |
| 157 | node_postprocessors=node_postprocessors |
| 158 | ) |
| 159 | return query_engine |
| 160 | |
| 161 | def load_node_postprocessors(self): |
| 162 | return [] |
no test coverage detected