Hotfix textual inv logging (huggingface#2183)

isamu-isozaki · web-flow · commit d46d78c5846f · 2023-02-03T18:08:46.000+01:00
diff --git 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)

Original file line number	Diff line number	Diff line change
`@@ -783,6 +783,7 @@ def main():`
`783`	`783`	`pipeline = DiffusionPipeline.from_pretrained(`
`784`	`784`	`args.pretrained_model_name_or_path,`
`785`	`785`	`text_encoder=accelerator.unwrap_model(text_encoder),`
	`786`	`+ tokenizer=tokenizer,`
`786`	`787`	`revision=args.revision,`
`787`	`788`	`)`
`788`	`789`	`pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)`