[refactor] teacache refactor (#395)

Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>

vllm_omni/diffusion/hooks.py

@@ -38,13 +38,13 @@ class ModelHook:
     """Base class for model hooks that can override a module's forward."""
 
     def initialize_hook(self, module: nn.Module) -> nn.Module:
-        return module
+        raise NotImplementedError
 
     def new_forward(self, module: nn.Module, *args: Any, **kwargs: Any):
         raise NotImplementedError
 
     def reset_state(self, module: nn.Module) -> nn.Module:
-        return module
+        raise NotImplementedError
 
 
 @dataclass