Update src/diffusers/models/unets/unet_motion_model.py

Co-authored-by: YiYi Xu <yixu310@gmail.com>
Merge branch 'main' into animatediff-warning
2025-12-08 05:24:20 +08:00 · 2024-07-02 10:20:42 +05:30 · 2024-07-01 07:35:02 -10:00 · 2024-07-01 12:32:56 +00:00
1 changed files with 3 additions and 1 deletions
--- a/src/diffusers/models/unets/unet_motion_model.py
+++ b/src/diffusers/models/unets/unet_motion_model.py
@@ -597,7 +597,9 @@ class UNetMotionModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixin):
        if not config.get("num_attention_heads"):
            config["num_attention_heads"] = config["attention_head_dim"]

-        config = FrozenDict(config)
+        expected_kwargs, optional_kwargs = cls._get_signature_keys(cls)
+        config = FrozenDict({k: config.get(k) for k in config if k in expected_kwargs or k in optional_kwargs})
+        config["_class_name"] = cls.__name__
        model = cls.from_config(config)

        if not load_weights:
Author	SHA1	Message	Date
Dhruv Nair	a12d8d90e2	Update src/diffusers/models/unets/unet_motion_model.py Co-authored-by: YiYi Xu <yixu310@gmail.com>	2024-07-02 10:20:42 +05:30
YiYi Xu	5a2909734d	Merge branch 'main' into animatediff-warning	2024-07-01 07:35:02 -10:00
Dhruv Nair	5ce8e040aa	update	2024-07-01 12:32:56 +00:00