Merge remote-tracking branch 'origin/master' into v2.0.0a-api-fix

2024-03-07 08:42:45 +00:00 · 2024-03-07 08:42:45 +00:00 · 98740f3b87
parent d1791040f5 12a503b8b7
commit 98740f3b87
1 changed files with 4 additions and 0 deletions
--- a/motion_module.py
+++ b/motion_module.py
@ -76,6 +76,10 @@ class MotionWrapper(nn.Module):
    def is_adxl(self):
        return self.mm_type == MotionModuleType.AnimateDiffXL

+    @property
+    def is_hotshot(self):
+        return self.mm_type == MotionModuleType.HotShotXL
+

    @property
    def is_v2(self):