HavelockAI
/

bert-token-classifier

Token Classification

Model card Files Files and versions

permutans commited on Feb 13

Commit

02643c1

·

verified ·

1 Parent(s): 91ec653

Upload folder using huggingface_hub

Files changed (1) hide show

modeling_havelock.py +1 -1

modeling_havelock.py CHANGED Viewed

@@ -15,7 +15,7 @@ class MultiLabelCRF(nn.Module):
         self.start_transitions = nn.Parameter(torch.empty(num_types, 3))
         self.end_transitions = nn.Parameter(torch.empty(num_types, 3))
         # Placeholder — will be overwritten by loaded weights if present
-        self.register_buffer("emission_bias", None)
         self._reset_parameters()
     def _reset_parameters(self) -> None:

         self.start_transitions = nn.Parameter(torch.empty(num_types, 3))
         self.end_transitions = nn.Parameter(torch.empty(num_types, 3))
         # Placeholder — will be overwritten by loaded weights if present
+        self.register_buffer("emission_bias", torch.zeros(1, 1, 1, 3))
         self._reset_parameters()
     def _reset_parameters(self) -> None: