Skip to content

Commit 2cbcc24

Browse files
committed
fix scaling function for UNetMotionModel
1 parent b753866 commit 2cbcc24

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/loaders/peft.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@
3030

3131
_SET_ADAPTER_SCALE_FN_MAPPING = {
3232
"UNet2DConditionModel": _maybe_expand_lora_scales,
33+
"UNetMotionModel": _maybe_expand_lora_scales,
3334
"SD3Transformer2DModel": lambda model_cls, weights: weights,
34-
"UNetMotionModel": lambda model_cls, weights: weights,
3535
}
3636

3737

0 commit comments

Comments
 (0)