Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
0d96a894
"vscode:/vscode.git/clone" did not exist on "4279d8ca44eca22994406345d91a31fb89b142a7"
Unverified
Commit
0d96a894
authored
Dec 17, 2024
by
Aryan
Committed by
GitHub
Dec 17, 2024
Browse files
Fix copied from comment in Mochi lora loader (#10255)
update
parent
6fb94d51
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
src/diffusers/loaders/lora_pipeline.py
src/diffusers/loaders/lora_pipeline.py
+2
-2
No files found.
src/diffusers/loaders/lora_pipeline.py
View file @
0d96a894
...
...
@@ -3104,7 +3104,7 @@ class Mochi1LoraLoaderMixin(LoraBaseMixin):
)
@
classmethod
# Copied from diffusers.loaders.lora_pipeline.SD3LoraLoaderMixin.load_lora_into_transformer with SD3Transformer2DModel->
CogVideoX
Transformer3DModel
# Copied from diffusers.loaders.lora_pipeline.SD3LoraLoaderMixin.load_lora_into_transformer with SD3Transformer2DModel->
Mochi
Transformer3DModel
def
load_lora_into_transformer
(
cls
,
state_dict
,
transformer
,
adapter_name
=
None
,
_pipeline
=
None
,
low_cpu_mem_usage
=
False
):
...
...
@@ -3116,7 +3116,7 @@ class Mochi1LoraLoaderMixin(LoraBaseMixin):
A standard state dict containing the lora layer parameters. The keys can either be indexed directly
into the unet or prefixed with an additional `unet` which can be used to distinguish between text
encoder lora layers.
transformer (`
CogVideoX
Transformer3DModel`):
transformer (`
Mochi
Transformer3DModel`):
The Transformer model to load the LoRA layers into.
adapter_name (`str`, *optional*):
Adapter name to be used for referencing the loaded adapter model. If not specified, it will use
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment