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

ignore

Browse source
This commit is contained in:
pesser 2022-05-30 14:04:35 +00:00
parent 8d6c7e34f2
commit f8c559e862
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
ldm/data/laion.py
View file

@ -140,8 +140,11 @@ class WebDataModuleFromConfig(pl.LightningDataModule):
print(f'Dataset holding {len(dset.pipeline[0].urls)} shards') print(f'Dataset holding {len(dset.pipeline[0].urls)} shards')
def ignore_me(*args, **kwargs):
pass
dset = (dset dset = (dset
.decode('pil') .decode('pil', handler=ignore_me)
# .to_tuple("jpg;png;jpeg pickle cls hls") # .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_tuple(image_transforms,load_partial_from_config(nns_transform) if 'target' in nns_transform else identity,identity,identity)
.map_dict(**transform_dict) .map_dict(**transform_dict)