fixing

SAE/sae.py

@@ -106,6 +106,7 @@ class SparseAutoencoder(nn.Module):
 
     @spaces.GPU
     def encode(self, x):
+        self = self.to('cuda')
         x = x - self.pre_bias
         latents_pre_act = self.encoder(x) + self.latent_bias