wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-12 12:47:45 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

Unbork Radiance NeRF embedder init

Browse Source
This commit is contained in:
blepping 2025-09-09 15:50:33 -06:00
parent cc6e7d60fd
commit 0828916ef1
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/chroma_radiance/layers.py
View File

@ -42,7 +42,9 @@ class NerfEmbedder(nn.Module):
# A linear layer to project the concatenated input features and
# positional encodings to the final output dimension.
self.embedder = operations.Linear(in_channels + max_freqs**2, hidden_size_input, dtype=dtype, device=device)
self.embedder = nn.Sequential(
operations.Linear(in_channels + max_freqs**2, hidden_size_input, dtype=dtype, device=device)
)
@lru_cache(maxsize=4)
def fetch_pos(self, patch_size: int, device: torch.device, dtype: torch.dtype) -> torch.Tensor: