andreas/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Changes to the previous radiance commit. (#9851)

Browse Source
This commit is contained in:
comfyanonymous
2025-09-13 15:03:34 -07:00
committed by GitHub
parent c1297f4eb3
commit 80b7c9455b
5 changed files with 35 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/ldm/chroma_radiance/model.py
View File

@@ -306,8 +306,9 @@ class ChromaRadiance(Chroma):
params = self.radiance_get_override_params(transformer_options.get("chroma_radiance_options", {}))
h_len = ((h + (self.patch_size // 2)) // self.patch_size)
w_len = ((w + (self.patch_size // 2)) // self.patch_size)
h_len = (img.shape[-2] // self.patch_size)
w_len = (img.shape[-1] // self.patch_size)
img_ids = torch.zeros((h_len, w_len, 3), device=x.device, dtype=x.dtype)
img_ids[:, :, 1] = img_ids[:, :, 1] + torch.linspace(0, h_len - 1, steps=h_len, device=x.device, dtype=x.dtype).unsqueeze(1)
img_ids[:, :, 2] = img_ids[:, :, 2] + torch.linspace(0, w_len - 1, steps=w_len, device=x.device, dtype=x.dtype).unsqueeze(0)
@@ -325,4 +326,4 @@ class ChromaRadiance(Chroma):
transformer_options,
attn_mask=kwargs.get("attention_mask", None),
)
return self.forward_nerf(img, img_out, params)
return self.forward_nerf(img, img_out, params)[:, :, :h, :w]