From f8c559e8627e2767351ba71edf3eee89a7f51721 Mon Sep 17 00:00:00 2001 From: pesser Date: Mon, 30 May 2022 14:04:35 +0000 Subject: [PATCH] ignore --- ldm/data/laion.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/ldm/data/laion.py b/ldm/data/laion.py index 05fc245..24f8f1e 100644 --- a/ldm/data/laion.py +++ b/ldm/data/laion.py @@ -140,8 +140,11 @@ class WebDataModuleFromConfig(pl.LightningDataModule): print(f'Dataset holding {len(dset.pipeline[0].urls)} shards') + def ignore_me(*args, **kwargs): + pass + dset = (dset - .decode('pil') + .decode('pil', handler=ignore_me) # .to_tuple("jpg;png;jpeg pickle cls hls") # .map_tuple(image_transforms,load_partial_from_config(nns_transform) if 'target' in nns_transform else identity,identity,identity) .map_dict(**transform_dict)