original_docstring: 
          Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer}
          
docstring: Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer}
docstring_tokens: Prunes heads of the model . heads_to_prune : dict of { layer_num : list of heads to prune in this layer }
code: def _prune_heads(self, heads_to_prune):
          """
          Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer}
          """
          for layer, heads in heads_to_prune.items():
              self.h[layer].attn.prune_heads(heads)
code_tokens: def _prune_heads ( self , heads_to_prune ) : for layer , heads in heads_to_prune . items ( ) : self . h [ layer ] . attn . prune_heads ( heads )
short_docstring: Prunes heads of the model.
short_docstring_tokens: Prunes heads of the model .
comment: """
          Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer}
          """