Searched refs:replace_tiled_token_positional_embedding (Results 1 – 4 of 4) sorted by relevance
15 replace_tiled_token_positional_embedding,68 self.model = replace_tiled_token_positional_embedding(self.model)
15 replace_tiled_token_positional_embedding,227 m = replace_tiled_token_positional_embedding(m)
9 replace_tiled_token_positional_embedding,
690 def replace_tiled_token_positional_embedding(model: nn.Module) -> nn.Module: function726 replace_tiled_token_positional_embedding(module)