def zero_parameters(self):
        """Zero out all parameters in embedding."""
        self.word_embeddings.weight.data.fill_(0)
        self.word_embeddings.weight.shared = True
        self.position_embeddings.weight.data.fill_(0)
        self.position_embeddings.weight.shared = True
Zero out all parameters in embedding.
training/DeepSpeed-Domino/megatron/core/models/gpt/gpt_embedding.py