two minor fixes (#335)

This commit is contained in:
Awni Hannun
2024-01-18 14:18:13 -08:00
committed by GitHub
parent d8680a89f9
commit bcc9fc3581
3 changed files with 12 additions and 3 deletions

View File

@@ -69,5 +69,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)