mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
simplified code a bit
This commit is contained in:
parent
288e31fc60
commit
a49d546125
@ -108,8 +108,7 @@ class Txt2Mask(object):
|
||||
padding=True,
|
||||
return_tensors='pt')
|
||||
outputs = self.model(**inputs)
|
||||
preds = outputs.logits
|
||||
heatmap = torch.sigmoid(preds)
|
||||
heatmap = torch.sigmoid(outputs.logits)
|
||||
return SegmentedGrayscale(image, heatmap)
|
||||
|
||||
def _scale_and_crop(self, image:Image)->Image:
|
||||
|
Loading…
Reference in New Issue
Block a user