From b2664e807ef616896f0785b8c88cd4f4352c501b Mon Sep 17 00:00:00 2001
From: Kevin Turner <83819+keturn@users.noreply.github.com>
Date: Sun, 4 Dec 2022 20:12:47 -0800
Subject: [PATCH] fix debug_image to not crash with non-RGB images.

---
 ldm/util.py | 24 ++++++++++--------------
 1 file changed, 10 insertions(+), 14 deletions(-)

diff --git a/ldm/util.py b/ldm/util.py
index ae28edb96a..b653c8e7db 100644
--- a/ldm/util.py
+++ b/ldm/util.py
@@ -1,17 +1,13 @@
 import importlib
-
-import torch
-import numpy as np
 import math
-from collections import abc
-from einops import rearrange
-from functools import partial
-
 import multiprocessing as mp
-from threading import Thread
-from queue import Queue
-
+from collections import abc
 from inspect import isfunction
+from queue import Queue
+from threading import Thread
+
+import numpy as np
+import torch
 from PIL import Image, ImageDraw, ImageFont
 
 
@@ -221,7 +217,7 @@ def rand_perlin_2d(shape, res, device, fade = lambda t: 6*t**5 - 15*t**4 + 10*t*
     grid = torch.stack(torch.meshgrid(torch.arange(0, res[0], delta[0]), torch.arange(0, res[1], delta[1]), indexing='ij'), dim = -1).to(device) % 1
 
     rand_val = torch.rand(res[0]+1, res[1]+1)
-    
+
     angles = 2*math.pi*rand_val
     gradients = torch.stack((torch.cos(angles), torch.sin(angles)), dim = -1).to(device)
 
@@ -249,8 +245,8 @@ def ask_user(question: str, answers: list):
 def debug_image(debug_image, debug_text, debug_show=True, debug_result=False, debug_status=False ):
     if not debug_status:
         return
-    
-    image_copy = debug_image.copy()
+
+    image_copy = debug_image.copy().convert("RGBA")
     ImageDraw.Draw(image_copy).text(
         (5, 5),
         debug_text,
@@ -261,4 +257,4 @@ def debug_image(debug_image, debug_text, debug_show=True, debug_result=False, de
         image_copy.show()
 
     if debug_result:
-        return image_copy
\ No newline at end of file
+        return image_copy