From b23664c794a229d4b2f6dc782b0442f2f81c650c Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Fri, 3 Feb 2023 17:32:35 -0500 Subject: [PATCH 1/2] registration of mask images was off due to typo - Problem found and fixed by @spezialspezial - Closes #2470 --- ldm/invoke/txt2mask.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ldm/invoke/txt2mask.py b/ldm/invoke/txt2mask.py index ccb7b1c604..d63b26df43 100644 --- a/ldm/invoke/txt2mask.py +++ b/ldm/invoke/txt2mask.py @@ -101,10 +101,10 @@ class Txt2Mask(object): image = ImageOps.exif_transpose(image) img = self._scale_and_crop(image) - img = transform(img).unsqueeze(0) + # img = transform(img).unsqueeze(0) inputs = self.processor(text=[prompt], - images=[image], + images=[img], padding=True, return_tensors='pt') outputs = self.model(**inputs) From d569c9dec62c7b217d21b57787ad417dca6e72e1 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Fri, 3 Feb 2023 17:35:35 -0500 Subject: [PATCH 2/2] remove dead code --- ldm/invoke/txt2mask.py | 1 - 1 file changed, 1 deletion(-) diff --git a/ldm/invoke/txt2mask.py b/ldm/invoke/txt2mask.py index d63b26df43..ab9f94549b 100644 --- a/ldm/invoke/txt2mask.py +++ b/ldm/invoke/txt2mask.py @@ -101,7 +101,6 @@ class Txt2Mask(object): image = ImageOps.exif_transpose(image) img = self._scale_and_crop(image) - # img = transform(img).unsqueeze(0) inputs = self.processor(text=[prompt], images=[img],