Commit f5afd706 authored by novelailab's avatar novelailab

dont postprocess dalle output

parent 38b2a77f
......@@ -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
......
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