Spaces:

PlayfulTechnology
/

QARAC

Build error

PeteBleackley commited on Apr 5, 2024

Commit

4808422

1 Parent(s): 7b60f84

Diagnostics

Files changed (1) hide show

qarac/models/layers/FactorizedMatrixMultiplication.py CHANGED Viewed

@@ -20,4 +20,6 @@ class FactorizedMatrixMultiplication(torch.nn.Module):
         self.matrix = torch.tensordot(self.left,self.right,1)
     def forward(self,X):
         return torch.einsum('ij,klj->kli',self.matrix,X)

         self.matrix = torch.tensordot(self.left,self.right,1)
     def forward(self,X):
+        print(X.device)
+        print(self.matrix.device)
         return torch.einsum('ij,klj->kli',self.matrix,X)