Skip to content

GitLab

Unverified Commit 61826040 authored by Long(Tony) Lian's avatar Long(Tony) Lian Committed by GitHub
Browse files

LLMGroundedDiffusionPipeline: inherit from DiffusionPipeline and fix peft (#6023) 

* LLMGroundedDiffusionPipeline: inherit from DiffusionPipeline and fix peft

* Use main in the revision in the examples

* Add "Copied from" statements in comments

* Fix formatting with ruff
parent dadd55fb
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 615 additions and 25 deletions
examples/community/README.md
View file @ 61826040
...@@ -78,6 +78,7 @@ from diffusers import DiffusionPipeline ...@@ -78,6 +78,7 @@ from diffusers import DiffusionPipeline
pipe = DiffusionPipeline.from_pretrained( pipe = DiffusionPipeline.from_pretrained(
"longlian/lmd_plus", "longlian/lmd_plus",
custom_pipeline="llm_grounded_diffusion", custom_pipeline="llm_grounded_diffusion",
custom_revision="main",
variant="fp16", torch_dtype=torch.float16 variant="fp16", torch_dtype=torch.float16
) )
pipe.enable_model_cpu_offload() pipe.enable_model_cpu_offload()
......
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