Install sub directories with folders correctly, ensure consistent dtype of tensors in flux pipeline and vae

This commit is contained in:
Brandon Rising
2024-08-19 13:59:44 -04:00
parent be6cb2c07c
commit 115f350f6f
5 changed files with 7 additions and 9 deletions

View File

@ -69,7 +69,7 @@ def filter_files(
# limit search to subfolder if requested
if subfolder:
subfolder = root / subfolder
paths = [x for x in paths if x.parent == Path(subfolder)]
paths = [x for x in paths if Path(subfolder) in x.parents]
# _filter_by_variant uniquifies the paths and returns a set
return sorted(_filter_by_variant(paths, variant))