mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-14 11:07:24 +08:00
Update comfy/ldm/common_dit.py
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
This commit is contained in:
parent
cfff1c07c4
commit
f186027ad0
@ -16,7 +16,7 @@ def pad_to_patch_size(img, patch_size=(2, 2), padding_mode="circular"):
|
||||
|
||||
rms_norm = comfy.rmsnorm.rms_norm
|
||||
|
||||
def process_img(x, index=0, h_offset=0, w_offset=0, patch_size=(2, 2), transformer_options={}):
|
||||
def process_img(x, index=0, h_offset=0, w_offset=0, patch_size=2, transformer_options={}):
|
||||
bs, c, h, w = x.shape
|
||||
x = pad_to_patch_size(x, (patch_size, patch_size))
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user