Hotfix textual inv logging (#2183)

2026-01-27 17:22:53 +03:00 · 2023-02-04 02:08:46 +09:00
parent 05168e5d83
commit d46d78c584
1 changed files with 1 additions and 0 deletions
--- a/examples/textual_inversion/textual_inversion.py
+++ b/examples/textual_inversion/textual_inversion.py
@@ -783,6 +783,7 @@ def main():
            pipeline = DiffusionPipeline.from_pretrained(
                args.pretrained_model_name_or_path,
                text_encoder=accelerator.unwrap_model(text_encoder),
+                tokenizer=tokenizer,
                revision=args.revision,
            )
            pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)