Method
__init__
(
self,
tokenizer,
tokenizer_type,
lang_encoder,
lang_projection,
max_token_num,
queue_operator,
)
Source from the content-addressed store, hash-verified
| 20 | |
| 21 | @configurable |
| 22 | def __init__( |
| 23 | self, |
| 24 | tokenizer, |
| 25 | tokenizer_type, |
| 26 | lang_encoder, |
| 27 | lang_projection, |
| 28 | max_token_num, |
| 29 | queue_operator, |
| 30 | ): |
| 31 | super().__init__() |
| 32 | # seg |
| 33 | self.tokenizer = tokenizer |
| 34 | self.tokenizer_type = tokenizer_type |
| 35 | self.lang_encoder = lang_encoder |
| 36 | self.lang_proj = lang_projection |
| 37 | self.max_token_num = max_token_num |
| 38 | self.logit_scale = nn.Parameter(torch.ones([])) |
| 39 | |
| 40 | # captioning & retrieval |
| 41 | for key, value in queue_operator.items(): |
| 42 | self.register_buffer(key, value) |
| 43 | |
| 44 | |
| 45 | @classmethod |
Callers
nothing calls this directly
Tested by
no test coverage detected