diff --git a/motion_module.py b/motion_module.py
index e6055c1..4fffd52 100644
--- 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):