Skip to content

Commit 19bd9b8

Browse files
committed
Unbork Radiance NeRF embedder init
1 parent 265be98 commit 19bd9b8

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

comfy/ldm/chroma_radiance/layers.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,9 @@ def __init__(
4242

4343
# A linear layer to project the concatenated input features and
4444
# positional encodings to the final output dimension.
45-
self.embedder = operations.Linear(in_channels + max_freqs**2, hidden_size_input, dtype=dtype, device=device)
45+
self.embedder = nn.Sequential(
46+
operations.Linear(in_channels + max_freqs**2, hidden_size_input, dtype=dtype, device=device)
47+
)
4648

4749
@lru_cache(maxsize=4)
4850
def fetch_pos(self, patch_size: int, device: torch.device, dtype: torch.dtype) -> torch.Tensor:

0 commit comments

Comments
 (0)