mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-22 11:57:44 +08:00
Update model_base.py
This commit is contained in:
parent
5fe386157a
commit
f8fd20a2f3
@ -1570,8 +1570,6 @@ class HunyuanVideo15(HunyuanVideo):
|
||||
if mask.shape[-3] < noise.shape[-3]:
|
||||
mask = torch.nn.functional.pad(mask, (0, 0, 0, 0, 0, noise.shape[-3] - mask.shape[-3]), mode='constant', value=0)
|
||||
mask = utils.resize_to_batch_size(mask, noise.shape[0])
|
||||
print("image.shape:", image.shape)
|
||||
print("mask.shape:", mask.shape)
|
||||
|
||||
return torch.cat((image, mask), dim=1)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user