add register_buffer/parameter to NamedModule class

gptqmodel/looper/named_module.py

@@ -3,11 +3,12 @@
 # SPDX-License-Identifier: Apache-2.0
 # Contact: qubitium@modelcloud.ai, x.com/qubitium
 
-from typing import Any
+from typing import Any, Optional
 
 import torch
 import transformers
-from torch import nn
+from torch import nn, Tensor
+from torch.nn import Parameter
 from torch.nn.modules.conv import _ConvNd
 
 
@@ -55,6 +56,18 @@ def __init__(self, module: torch.nn.Module, name: str, full_name:str, layer_inde
             "out_features": out_features,
         })
 
+    def register_buffer(
+        self, name: str, tensor: Optional[Tensor], persistent: bool = True
+    ) -> None:
+        return self.module.register_buffer(name, tensor, persistent)
+
+    def unregister_buffer(self, name: str):
+        del self.module._buffers[name]
+        delattr(self.module, "mask")
+
+    def register_parameter(self, name: str, param: Optional[Parameter]) -> None:
+        return self.module.register_parameter(name, param)
+
     # return stats for mo
     # def stats(self) -> Dict[str, float]:
     #     # -1 means no stats have yet to gathered for the stat property