Skip to content

GitLab

"src/git@developer.sourcefind.cn:OpenDAS/lmdeploy.git" did not exist on "e2d98ddce2d31dcc95c58604d9ddae8232344a1d"
Unverified Commit d46d78c5 authored by Isamu Isozaki's avatar Isamu Isozaki Committed by GitHub
Browse files

Hotfix textual inv logging (#2183)

parent 05168e5d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
examples/textual_inversion/textual_inversion.py
View file @ d46d78c5
...@@ -783,6 +783,7 @@ def main(): ...@@ -783,6 +783,7 @@ def main():
pipeline = DiffusionPipeline.from_pretrained( pipeline = DiffusionPipeline.from_pretrained(
args.pretrained_model_name_or_path, args.pretrained_model_name_or_path,
text_encoder=accelerator.unwrap_model(text_encoder), text_encoder=accelerator.unwrap_model(text_encoder),
tokenizer=tokenizer,
revision=args.revision, revision=args.revision,
) )
pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config) pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment