From d46d78c5846f88552abdfdd67dd6e704c0c47a60 Mon Sep 17 00:00:00 2001 From: Isamu Isozaki Date: Sat, 4 Feb 2023 02:08:46 +0900 Subject: [PATCH] Hotfix textual inv logging (#2183) --- examples/textual_inversion/textual_inversion.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/textual_inversion/textual_inversion.py b/examples/textual_inversion/textual_inversion.py index 1c4bd235e6..d41198c369 100644 --- 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)