Remove suspicious cast

hydra_node/models.py

@@ -91,7 +91,7 @@ def encode_image(image, model):
     if isinstance(image, np.ndarray):
         image = torch.from_numpy(image)
 
-    dtype = image.dtype
+    #dtype = image.dtype
     image = image.to(torch.float32)
     #gets image as numpy array and returns as tensor
     def preprocess_vqgan(x):
@@ -102,7 +102,7 @@ def encode_image(image, model):
     image = image.permute(2, 0, 1).unsqueeze(0).float().cuda()
     image = preprocess_vqgan(image)
     image = model.encode(image).sample()
-    image = image.to(dtype)
+    #image = image.to(dtype)
 
     return image
 
@@ -723,4 +723,4 @@ class EmbedderModel(nn.Module):
 
         results = results[:-len(found)]
         results = found + results
-        return results
\ No newline at end of file
+        return results