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

stop crashes on non-square images

Browse Source
This commit is contained in:
Lincoln Stein
2022-10-25 13:17:06 -04:00
parent dd07392045
commit 4352eb6628
3 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ldm/invoke/generator/omnibus.py
View File

@ -71,6 +71,8 @@ class Omnibus(Img2Img,Txt2Img):
mask_image = torch.ones(1, 1, height, width, device=self.model.device)
masked_image = init_image
height = init_image.shape[2]
width = init_image.shape[3]
model = self.model
def make_image(x_T):
@ -88,7 +90,6 @@ class Omnibus(Img2Img,Txt2Img):
)
c = model.cond_stage_model.encode(batch["txt"])
c_cat = list()
for ck in model.concat_keys:
cc = batch[ck].float()