Skip to content

Commit

Permalink
do not process vae split without preprocessing
Browse files Browse the repository at this point in the history
  • Loading branch information
bghira committed Apr 9, 2024
1 parent 06e2721 commit f2fc3bf
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions helpers/data_backend/factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -661,8 +661,10 @@ def configure_multi_databackend(
init_backend["metadata_backend"].load_image_metadata()
accelerator.wait_for_everyone()

if "vae" not in args.skip_file_discovery and "vae" not in backend.get(
"skip_file_discovery", ""
if (
args.vae_cache_preprocess
and "vae" not in args.skip_file_discovery
and "vae" not in backend.get("skip_file_discovery", "")
):
init_backend["vaecache"].split_cache_between_processes()
if args.vae_cache_preprocess:
Expand Down

0 comments on commit f2fc3bf

Please sign in to comment.