Method__init__(self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.)
svrm/ldm/modules/attention.py:211
Method__init__(self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.)
svrm/ldm/modules/attention.py:247
Method__init__(self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.0)
svrm/ldm/modules/attention.py:286
Method__init__(self, dim, n_heads, d_head, dropout=0., context_dim=None, gated_ff=True, checkpoint=True,
di
svrm/ldm/modules/attention.py:336
Method__init__(self, dim, n_heads, d_head, dropout=0., context_dim=None, gated_ff=True, \
checkpoint=True)
svrm/ldm/modules/attention.py:394
Method__init__(self, query_dim, n_heads, d_head, depth=1, dropout=0., context_dim=None, triplane_size=64)
svrm/ldm/modules/attention.py:435
Method__init__(
self,
in_dim,
out_dim,
use_bn=False,
nlayers=3,
hidden_dim=2
svrm/ldm/modules/encoders/dinov2/layers/dino_head.py:13
Method__init__(
self,
img_size: Union[int, Tuple[int, int]] = 224,
patch_size: Union[int, Tuple[int,
svrm/ldm/modules/encoders/dinov2/layers/patch_embed.py:37
Method__init__(
self, v_pos: Float[Tensor, "Nv 3"], t_pos_idx: Integer[Tensor, "Nf 3"], v_rgb: Integer[Tensor, "Nf 3
svrm/ldm/modules/rendering_neus/mesh.py:13
Method__init__(self, dataset, rank=0, num_replicas=1, shuffle=True, seed=0, window_size=0.5)
svrm/ldm/modules/rendering_neus/third_party/misc.py:114
Methodantialias(
self,
color: Float[Tensor, "B H W C"],
rast: Float[Tensor, "B H W 4"],
pos:
svrm/ldm/modules/rendering_neus/rasterize.py:49