diff options
author | zhaohu xing <920232796@qq.com> | 2022-11-30 06:56:12 +0000 |
---|---|---|
committer | zhaohu xing <920232796@qq.com> | 2022-11-30 06:56:12 +0000 |
commit | 52cc83d36b7663a77b79fd2258d2ca871af73e55 (patch) | |
tree | 5c31e75a3934327331d5636bd6ef1420c3ba32fe /ldm/data/base.py | |
parent | a39a57cb1f5964d9af2b541f7b352576adeeac0f (diff) | |
download | stable-diffusion-webui-gfx803-52cc83d36b7663a77b79fd2258d2ca871af73e55.tar.gz stable-diffusion-webui-gfx803-52cc83d36b7663a77b79fd2258d2ca871af73e55.tar.bz2 stable-diffusion-webui-gfx803-52cc83d36b7663a77b79fd2258d2ca871af73e55.zip |
fix bugs
Signed-off-by: zhaohu xing <920232796@qq.com>
Diffstat (limited to 'ldm/data/base.py')
-rw-r--r-- | ldm/data/base.py | 23 |
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 |