fixing

SAE/sae.py

@@ -107,10 +107,7 @@ class SparseAutoencoder(nn.Module):
 
     @spaces.GPU
     def encode(self, x):
-
-        print(x.device)
-        print(self.pre_bias.device)
-        x = x - self.pre_bias
+        x = x.to('cuda') - self.pre_bias
         latents_pre_act = self.encoder(x) + self.latent_bias
 
         vals, inds = torch.topk(