aboutsummaryrefslogtreecommitdiff
path: root/ldm/data/base.py
diff options
context:
space:
mode:
authorzhaohu xing <920232796@qq.com>2022-11-30 14:56:12 +0800
committerzhaohu xing <920232796@qq.com>2022-11-30 14:56:12 +0800
commit52cc83d36b7663a77b79fd2258d2ca871af73e55 (patch)
tree5c31e75a3934327331d5636bd6ef1420c3ba32fe /ldm/data/base.py
parenta39a57cb1f5964d9af2b541f7b352576adeeac0f (diff)
fix bugs
Signed-off-by: zhaohu xing <920232796@qq.com>
Diffstat (limited to 'ldm/data/base.py')
-rw-r--r--ldm/data/base.py23
1 files changed, 0 insertions, 23 deletions
diff --git a/ldm/data/base.py b/ldm/data/base.py
deleted file mode 100644
index b196c2f7..00000000
--- a/ldm/data/base.py
+++ /dev/null
@@ -1,23 +0,0 @@
-from abc import abstractmethod
-from torch.utils.data import Dataset, ConcatDataset, ChainDataset, IterableDataset
-
-
-class Txt2ImgIterableBaseDataset(IterableDataset):
- '''
- Define an interface to make the IterableDatasets for text2img data chainable
- '''
- def __init__(self, num_records=0, valid_ids=None, size=256):
- super().__init__()
- self.num_records = num_records
- self.valid_ids = valid_ids
- self.sample_ids = valid_ids
- self.size = size
-
- print(f'{self.__class__.__name__} dataset contains {self.__len__()} examples.')
-
- def __len__(self):
- return self.num_records
-
- @abstractmethod
- def __iter__(self):
- pass \ No newline at end of file