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)