neulus/cloud-removal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

things

Browse Source
This commit is contained in:
neulus
2025-10-13 23:14:44 +09:00
parent c47d91a349
commit 3b03453e5d
28 changed files with 700 additions and 208 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/model/dino.py
View File

@@ -373,7 +373,7 @@ class DINOv3ViTModel(nn.Module):
self.config = config
self.embeddings = DINOv3ViTEmbeddings(config)
self.rope_embeddings = DINOv3ViTRopePositionEmbedding(config)
self.layers = nn.ModuleList(
self.layer = nn.ModuleList(
[DINOv3ViTLayer(config) for _ in range(config.num_hidden_layers)]
)
self.norm = nn.LayerNorm(config.hidden_size, eps=config.layer_norm_eps)
@@ -420,7 +420,7 @@ class DINOv3ViTModel(nn.Module):
position_embeddings = self.rope_embeddings(pixel_values)
latents = []
for i, layer_module in enumerate(self.layers):
for i, layer_module in enumerate(self.layer):
layer_head_mask = head_mask[i] if head_mask is not None else None
hidden_states = layer_module(
hidden_states,