fix is_hotshot (#453)

2024-03-07 00:14:17 -06:00 · 2024-03-07 00:14:17 -06:00 · 12a503b8b7
parent c92c72b5e5
commit 12a503b8b7
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):