(
self,
tokenizer,
tokenizer_type,
lang_encoder,
lang_projection,
max_token_num,
)
| 14 | |
| 15 | @configurable |
| 16 | def __init__( |
| 17 | self, |
| 18 | tokenizer, |
| 19 | tokenizer_type, |
| 20 | lang_encoder, |
| 21 | lang_projection, |
| 22 | max_token_num, |
| 23 | ): |
| 24 | super().__init__() |
| 25 | self.tokenizer = tokenizer |
| 26 | self.tokenizer_type = tokenizer_type |
| 27 | self.lang_encoder = lang_encoder |
| 28 | self.lang_proj = lang_projection |
| 29 | self.max_token_num = max_token_num |
| 30 | self.logit_scale = nn.Parameter(torch.ones([])) |
| 31 | |
| 32 | @classmethod |
| 33 | def from_config(cls, cfg): |
nothing calls this directly
no outgoing calls
no test coverage detected