This commit is contained in:
James Betker 2021-11-07 18:52:05 -07:00
parent 18b1de9b2c
commit 722d3dbdc2

View File

@ -590,7 +590,9 @@ def load_paths_from_cache(paths, cache_path, exclusion_list=[]):
output.extend(find_files_of_type('img', p, qualifier=is_audio_file)[0])
if exclusion_list is not None and len(exclusion_list) > 0:
print(f"Removing exclusion lists..")
before = len(output)
output = filter(lambda p: p not in exclusion_list, output)
print(f"Excluded {before-len(output)} files.")
print("Done.")
torch.save(output, cache_path)
return output