Fix SD image conversion (#266)

This commit is contained in:
Awni Hannun
2024-01-09 08:41:31 -08:00
committed by GitHub
parent 6e6eff326e
commit 6759dfddf1

View File

@@ -52,5 +52,5 @@ if __name__ == "__main__":
x = (x * 255).astype(mx.uint8)
# Save them to disc
im = Image.fromarray(x.__array__())
im = Image.fromarray(np.array(x))
im.save(args.output)