Skip to content

H
Hydra Node Http
Commit 4fe25b9c authored by novelailab's avatar novelailab
Browse files
Options

Remove suspicious cast

parent 7f127cf5
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
hydra_node/models.py
View file @ 4fe25b9c
...@@ -91,7 +91,7 @@ def encode_image(image, model): ...@@ -91,7 +91,7 @@ def encode_image(image, model):
if isinstance(image, np.ndarray): if isinstance(image, np.ndarray):
image = torch.from_numpy(image) image = torch.from_numpy(image)
dtype = image.dtype #dtype = image.dtype
image = image.to(torch.float32) image = image.to(torch.float32)
#gets image as numpy array and returns as tensor #gets image as numpy array and returns as tensor
def preprocess_vqgan(x): def preprocess_vqgan(x):
...@@ -102,7 +102,7 @@ def encode_image(image, model): ...@@ -102,7 +102,7 @@ def encode_image(image, model):
image = image.permute(2, 0, 1).unsqueeze(0).float().cuda() image = image.permute(2, 0, 1).unsqueeze(0).float().cuda()
image = preprocess_vqgan(image) image = preprocess_vqgan(image)
image = model.encode(image).sample() image = model.encode(image).sample()
image = image.to(dtype) #image = image.to(dtype)
return image return image
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment