as uint8

deff8bc2 · novelailab · 9b4d7040 · deff8bc2
Commit deff8bc2 authored Aug 04, 2022 by novelailab
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

hydra_node/models.py hydra_node/models.py +2 -1

No files found.
--- a/hydra_node/models.py
+++ b/hydra_node/models.py
@@ -147,7 +147,8 @@ class DalleMiniModel(nn.Module):
            supercondition_factor=request.scale,
            is_verbose=False
        )
-        images = images.to('cpu').numpy().transpose(0, 2, 3, 1)
+        images = images.to('cpu').numpy().transpose(0, 2, 3, 1) * 255.
+        images = images.astype(np.uint8)
        images = np.ascontiguousarray(images)
        return images