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
"git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "0cf4b8a9ec35ba0048051d0bc9d0ca8de5f085fe"
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):
...
@@ -3104,7 +3104,7 @@ class Mochi1LoraLoaderMixin(LoraBaseMixin):
)
)
@
classmethod
@
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
(
def
load_lora_into_transformer
(
cls
,
state_dict
,
transformer
,
adapter_name
=
None
,
_pipeline
=
None
,
low_cpu_mem_usage
=
False
cls
,
state_dict
,
transformer
,
adapter_name
=
None
,
_pipeline
=
None
,
low_cpu_mem_usage
=
False
):
):
...
@@ -3116,7 +3116,7 @@ class Mochi1LoraLoaderMixin(LoraBaseMixin):
...
@@ -3116,7 +3116,7 @@ class Mochi1LoraLoaderMixin(LoraBaseMixin):
A standard state dict containing the lora layer parameters. The keys can either be indexed directly
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
into the unet or prefixed with an additional `unet` which can be used to distinguish between text
encoder lora layers.
encoder lora layers.
transformer (`
CogVideoX
Transformer3DModel`):
transformer (`
Mochi
Transformer3DModel`):
The Transformer model to load the LoRA layers into.
The Transformer model to load the LoRA layers into.
adapter_name (`str`, *optional*):
adapter_name (`str`, *optional*):
Adapter name to be used for referencing the loaded adapter model. If not specified, it will use
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