From 6759dfddf1b899418143f84f9cf2a3dd7ad531a1 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Tue, 9 Jan 2024 08:41:31 -0800 Subject: [PATCH] Fix SD image conversion (#266) --- stable_diffusion/txt2image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stable_diffusion/txt2image.py b/stable_diffusion/txt2image.py index a487cd22..877acae7 100644 --- a/stable_diffusion/txt2image.py +++ b/stable_diffusion/txt2image.py @@ -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)