diff --git a/ops.py b/ops.py index 2b9be44..cea313e 100644 --- a/ops.py +++ b/ops.py @@ -117,6 +117,7 @@ class GGMLLayer(torch.nn.Module): # dequantize tensor while patches load weight = dequantize_tensor(tensor, dtype, self.dequant_dtype) + weight.__class__ = torch.Tensor # prevent propagating custom tensor class # apply patches if patch_list: