Hotfix textual inv logging (#2183)

d46d78c5 · Isamu Isozaki · GitHub · 05168e5d · d46d78c5
Unverified Commit d46d78c5 authored Feb 04, 2023 by Isamu Isozaki Committed by GitHub Feb 03, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

examples/textual_inversion/textual_inversion.py examples/textual_inversion/textual_inversion.py +1 -0

No files found.
--- 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)