LoRA issue workaround

Proposed by @ahiser1117 in #37
2024-08-18 04:29:58 +02:00
parent 4e848f7c95
commit 724b2d8559
1 changed files with 2 additions and 0 deletions
@@ -34,6 +34,8 @@ class GGMLTensor(torch.Tensor):

    @property
    def shape(self):
+        if not hasattr(self, "tensor_shape"):
+            self.tensor_shape = self.size() 
        return self.tensor_shape

 class GGMLLayer(torch.nn.Module):