remove cuda dependance for inference

utils.py

@@ -116,8 +116,8 @@ class VariationalEncoder(nn.Module):
         self.mu = nn.Linear(latent_size, latent_size)
         self.sigma = nn.Linear(latent_size, latent_size)
         self.N = torch.distributions.Normal(0, 1)
-        self.N.loc = self.N.loc.cuda()
-        self.N.scale = self.N.scale.cuda()
+        # self.N.loc = self.N.loc.cuda()
+        # self.N.scale = self.N.scale.cuda()
         self.kl = 0
     
     def forward(self, x):