From de7a003484d7e0b02322ff9e22225b2ff7e1f77e Mon Sep 17 00:00:00 2001 From: rromb Date: Tue, 31 May 2022 15:21:55 +0200 Subject: [PATCH] one more handler --- ldm/data/laion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ldm/data/laion.py b/ldm/data/laion.py index 3593545..4c38a4c 100644 --- a/ldm/data/laion.py +++ b/ldm/data/laion.py @@ -150,7 +150,7 @@ class WebDataModuleFromConfig(pl.LightningDataModule): dset = (dset .decode('pil', handler=warn_and_continue) - .map_dict(**transform_dict) + .map_dict(**transform_dict, handler=wds.warn_and_continue) .batched(self.batch_size, partial=False, collation_fn=dict_collation_fn) )