Restore text projection default OOV policy

This commit is contained in:
Shay 2026-05-13 13:14:41 -07:00
parent 58092112c3
commit 2e842341b3

View file

@ -50,7 +50,11 @@ class TextProjectionHead:
modality: Modality = Modality.TEXT
embedding_dim: int = CL41_DIM
def __init__(self, vocabulary: ModalityVocabulary, oov_policy: OOVPolicy) -> None:
def __init__(
self,
vocabulary: ModalityVocabulary,
oov_policy: OOVPolicy = OOVPolicy.TAGGED_FALLBACK,
) -> None:
self._vocab = vocabulary
self._oov_policy = oov_policy