fix error on pool_workaround in sdxl TE training ref #994

2026-04-06 13:47:06 +00:00 · 2023-12-10 09:18:33 +09:00
parent 912dca8f65
commit 42750f7846
4 changed files with 30 additions and 25 deletions
--- a/sdxl_train_textual_inversion.py
+++ b/sdxl_train_textual_inversion.py
@@ -64,6 +64,7 @@ class SdxlTextualInversionTrainer(train_textual_inversion.TextualInversionTraine
                text_encoders[0],
                text_encoders[1],
                None if not args.full_fp16 else weight_dtype,
+                accelerator=accelerator,
            )
        return encoder_hidden_states1, encoder_hidden_states2, pool2