one more handler

2022-05-31 15:21:55 +02:00 · 2022-05-31 15:21:55 +02:00 · de7a003484
commit de7a003484
parent 1eb0b88a8d
1 changed files with 1 additions and 1 deletions
--- 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)
                )