Fix VAE offload encode device mismatch in DreamBooth scripts#13417

Merged

sayakpaul merged 2 commits intohuggingface:mainfrom

azolotenkov:fix-flux2-vae-offload-device-mismatch

Apr 6, 2026

Commits on Apr 5, 2026

Fix VAE offload encode device mismatch in DreamBooth scripts
azolotenkov
committed

Commits on Apr 6, 2026

Merge branch 'main' into fix-flux2-vae-offload-device-mismatch
sayakpaul
authored