model.token_embedder.weight = nn.Parameter(torch.Tensor(weights["token_embedder"]["embedding"])) model.position_encoding.weight = nn.Parameter(torch.Tensor(weights ...