nodes/city96-ComfyUI_ExtraModels
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

HunYuanDiT Size cond wrong

Browse Source
This commit is contained in:
City
2024-05-25 17:44:28 +02:00
parent e5c15ba479
commit 904e5bcacf
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
HunYuanDiT/models/models.py
+2 -2
View File
@@ -373,8 +373,8 @@ class HunYuanDiT(nn.Module):
style = torch.as_tensor([0, 0] * (x.shape[0]//2), device=x.device) style = torch.as_tensor([0, 0] * (x.shape[0]//2), device=x.device)
# image size - todo: from cond # image size - todo: from cond
width = x.shape[3] width = x.shape[2]
height = x.shape[2] height = x.shape[3]
src_size_cond = (width//2*16, height//2*16) src_size_cond = (width//2*16, height//2*16)
size_cond = list(src_size_cond) + [width*8, height*8, 0, 0] size_cond = list(src_size_cond) + [width*8, height*8, 0, 0]
image_meta_size = torch.as_tensor([size_cond] * x.shape[0], device=x.device) image_meta_size = torch.as_tensor([size_cond] * x.shape[0], device=x.device)