registration of mask images was off due to typo

- Problem found and fixed by @spezialspezial
- Closes #2470
This commit is contained in:
Lincoln Stein 2023-02-03 17:32:35 -05:00
parent 76b0bdb6f9
commit b23664c794

View File

@ -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)