remove alpha_mask from device

spectra/spectra_gpu.py

@@ -268,7 +268,7 @@ def __init__(self, X, labels, adj_matrix, L, weights = None, lam = 10e-4, delta=
         self.n_cell_typesp1 = len(ct_order)
 
         #just need to construct these masks once
-        self.alpha_mask = torch.zeros((self.n,self.L_tot)).to(device)
+        self.alpha_mask = torch.zeros((self.n,self.L_tot))
         self.B_mask = torch.zeros((self.n_cell_typesp1, self.L_tot, self.L_tot)).to(device) #need to double check that this works
         self.factor_to_celltype = torch.zeros((self.L_tot, self.n_cell_typesp1)).to(device)