From 6813b0c19b377d6c2129670f0af43e35a59f3793 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Fri, 16 Aug 2024 04:55:34 -0500 Subject: [PATCH] Fix sd_type check for AD Per Block node --- animatediff/model_injection.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/animatediff/model_injection.py b/animatediff/model_injection.py index df5047c..0c4249f 100644 --- a/animatediff/model_injection.py +++ b/animatediff/model_injection.py @@ -1379,7 +1379,7 @@ def validate_model_compatibility_gen2(model: ModelPatcher, motion_model: MotionM def validate_per_block_compatibility(motion_model: MotionModelPatcher, all_per_blocks: AllPerBlocks): - if all_per_blocks is None: + if all_per_blocks is None or all_per_blocks.sd_type is None: return mm_info = motion_model.model.mm_info if all_per_blocks.sd_type != mm_info.sd_type: