Skip to content

Commit 66f8bd6

Browse files
authored
Fix vae_encodings_fn hash in train_text_to_image_sdxl.py (#7171)
Update train_text_to_image_sdxl.py
1 parent 64a8cd6 commit 66f8bd6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/text_to_image/train_text_to_image_sdxl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -895,7 +895,7 @@ def preprocess_train(examples):
895895
# fingerprint used by the cache for the other processes to load the result
896896
# details: https://github.com/huggingface/diffusers/pull/4038#discussion_r1266078401
897897
new_fingerprint = Hasher.hash(args)
898-
new_fingerprint_for_vae = Hasher.hash("vae")
898+
new_fingerprint_for_vae = Hasher.hash(vae_path)
899899
train_dataset = train_dataset.map(compute_embeddings_fn, batched=True, new_fingerprint=new_fingerprint)
900900
train_dataset = train_dataset.map(
901901
compute_vae_encodings_fn,

0 commit comments

Comments
 (0)