dont postprocess dalle output

f5afd706 · novelailab · 38b2a77f · f5afd706
Commit f5afd706 authored Aug 05, 2022 by novelailab
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

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