Skip to content

GitLab

Unverified Commit 66f8bd68 authored by Quentin Lhoest's avatar Quentin Lhoest Committed by GitHub
Browse files

Fix vae_encodings_fn hash in train_text_to_image_sdxl.py (#7171) 

Update train_text_to_image_sdxl.py
parent 64a8cd62
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/text_to_image/train_text_to_image_sdxl.py
View file @ 66f8bd68
...@@ -895,7 +895,7 @@ def main(args): ...@@ -895,7 +895,7 @@ def main(args):
# fingerprint used by the cache for the other processes to load the result # fingerprint used by the cache for the other processes to load the result
# details: https://github.com/huggingface/diffusers/pull/4038#discussion_r1266078401 # details: https://github.com/huggingface/diffusers/pull/4038#discussion_r1266078401
new_fingerprint = Hasher.hash(args) new_fingerprint = Hasher.hash(args)
new_fingerprint_for_vae = Hasher.hash("vae") new_fingerprint_for_vae = Hasher.hash(vae_path)
train_dataset = train_dataset.map(compute_embeddings_fn, batched=True, new_fingerprint=new_fingerprint) train_dataset = train_dataset.map(compute_embeddings_fn, batched=True, new_fingerprint=new_fingerprint)
train_dataset = train_dataset.map( train_dataset = train_dataset.map(
compute_vae_encodings_fn, compute_vae_encodings_fn,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment