src/transformers/models/realm/modeling_realm.py/RealmForOpenQA/block_embedding_to
class RealmForOpenQA:    def block_embedding_to(self, device):
        """Send `self.block_emb` to a specific device.

        Args:
            device (`str` or `torch.device`):
                The device to which `self.block_emb` will be sent.
        """

        self.block_emb = self.block_emb.to(device)