(self, map_fn, i, regenerate_cache=False, caching_batch_size=1)
| 438 | ds.cache_latents(map_fn, regenerate_cache=regenerate_cache, trust_cache=trust_cache, caching_batch_size=caching_batch_size) |
| 439 | |
| 440 | def cache_text_embeddings(self, map_fn, i, regenerate_cache=False, caching_batch_size=1): |
| 441 | print(f'caching text embeddings: {self.ar_frames}') |
| 442 | te_dataset = _cache_text_embeddings(self.metadata_dataset, map_fn, i, self.cache_dir, regenerate_cache, caching_batch_size) |
| 443 | for size_bucket_dataset in self.size_buckets: |
| 444 | size_bucket_dataset.add_text_embedding_dataset(te_dataset) |
| 445 | |
| 446 | |
| 447 | class DirectoryDataset: |
nothing calls this directly
no test coverage detected