Restore text projection default OOV policy
This commit is contained in:
parent
58092112c3
commit
2e842341b3
1 changed files with 5 additions and 1 deletions
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue