Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix debug_image to not crash with non-RGB images.

Browse Source
This commit is contained in:
Kevin Turner 2022-12-04 20:12:47 -08:00
parent 875312080d
commit b2664e807e
1 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
ldm/util.py
View File

@ -1,17 +1,13 @@
import importlib import importlib
import torch
import numpy as np
import math import math
from collections import abc
from einops import rearrange
from functools import partial
import multiprocessing as mp import multiprocessing as mp
from threading import Thread from collections import abc
from queue import Queue
from inspect import isfunction from inspect import isfunction
from queue import Queue
from threading import Thread
import numpy as np
import torch
from PIL import Image, ImageDraw, ImageFont from PIL import Image, ImageDraw, ImageFont
@ -250,7 +246,7 @@ def debug_image(debug_image, debug_text, debug_show=True, debug_result=False, de
if not debug_status: if not debug_status:
return return
image_copy = debug_image.copy() image_copy = debug_image.copy().convert("RGBA")
ImageDraw.Draw(image_copy).text( ImageDraw.Draw(image_copy).text(
(5, 5), (5, 5),
debug_text, debug_text,