Skip to content

Commit

Permalink
reduce error to local backends only
Browse files Browse the repository at this point in the history
  • Loading branch information
bghira committed Nov 16, 2024
1 parent 2bebe29 commit c6ce738
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion helpers/data_backend/factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -1054,7 +1054,9 @@ def configure_multi_databackend(args: dict, accelerator, text_encoders, tokenize
f"VAE image embed cache directory {backend.get('cache_dir_vae')} is the same as the text embed cache directory. This is not allowed, the trainer will get confused."
)

if vae_cache_dir is None or vae_cache_dir == "":
if backend["type"] == "local" and (
vae_cache_dir is None or vae_cache_dir == ""
):
raise ValueError(
f"VAE image embed cache directory {backend.get('cache_dir_vae')} is not set. This is required for the VAE image embed cache."
)
Expand Down

0 comments on commit c6ce738

Please sign in to comment.