r/stable-diffusion-finetune
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

one more handler

Browse source
This commit is contained in:
rromb 2022-05-31 15:21:55 +02:00
parent 1eb0b88a8d
commit de7a003484
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
ldm/data/laion.py
View file

@ -150,7 +150,7 @@ class WebDataModuleFromConfig(pl.LightningDataModule):
dset = (dset dset = (dset
.decode('pil', handler=warn_and_continue) .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, .batched(self.batch_size, partial=False,
collation_fn=dict_collation_fn) collation_fn=dict_collation_fn)
) )