zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-24 17:31:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix SD image conversion (#266)

Browse Source
This commit is contained in:
Awni Hannun 2024-01-09 08:41:31 -08:00 committed by GitHub
parent 6e6eff326e
commit 6759dfddf1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
stable_diffusion/txt2image.py
View File

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