<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def set_num_special_tokens(self, num_special_tokens):
 """ Update input and output embeddings with new embedding matrice Make sure we are sharing the embeddings """

        self.transformer.set_num_special_tokens(num_special_tokens)
        self.lm_head.set_embeddings_weights(self.transformer.tokens_embed.weight)