(self, media_chunk: "MTMDProcessor.MediaChunk")
| 10853 | ) |
| 10854 | |
| 10855 | def _load_cached_media_chunk(self, media_chunk: "MTMDProcessor.MediaChunk") -> bool: |
| 10856 | if self.embedding_cache is None: |
| 10857 | return False |
| 10858 | cached = self.embedding_cache.load(media_chunk.key) |
| 10859 | if cached is None or cached.embeddings.shape != ( |
| 10860 | media_chunk.n_tokens, |
| 10861 | self.n_embd_inp, |
| 10862 | ): |
| 10863 | return False |
| 10864 | media_chunk.embeddings = cached.embeddings |
| 10865 | return True |
| 10866 | |
| 10867 | def _save_media_chunk(self, media_chunk: "MTMDProcessor.MediaChunk") -> None: |
| 10868 | if self.embedding_cache is None or media_chunk.embeddings is None: |
no test coverage detected