mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-17 20:47:31 +08:00
Update comfy/ldm/chroma/model.py
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
This commit is contained in:
parent
aa68b226ea
commit
cfff1c07c4
@ -310,7 +310,7 @@ class Chroma(nn.Module):
|
|||||||
h = max(h, ref.shape[-2] + h_offset)
|
h = max(h, ref.shape[-2] + h_offset)
|
||||||
w = max(w, ref.shape[-1] + w_offset)
|
w = max(w, ref.shape[-1] + w_offset)
|
||||||
|
|
||||||
kontext, kontext_ids = comfy.ldm.common_dit.process_img(ref, index=index, h_offset=h_offset, w_offset=w_offset, patch_size=self.patch_size)
|
kontext, kontext_ids = comfy.ldm.common_dit.process_img(ref, index=index, h_offset=h_offset, w_offset=w_offset, patch_size=self.patch_size, transformer_options=transformer_options)
|
||||||
img = torch.cat([img, kontext], dim=1)
|
img = torch.cat([img, kontext], dim=1)
|
||||||
img_ids = torch.cat([img_ids, kontext_ids], dim=1)
|
img_ids = torch.cat([img_ids, kontext_ids], dim=1)
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user