Jonathan Tow commited on
Commit
1465d75
1 Parent(s): eee47ee

fix: update `LayerNorm` docstring

Browse files
Files changed (1) hide show
  1. modeling_stablelm_alpha.py +1 -1
modeling_stablelm_alpha.py CHANGED
@@ -51,7 +51,7 @@ def _expand_mask(mask: torch.Tensor, dtype: torch.dtype, tgt_len: Optional[int]
51
  class LayerNorm(nn.LayerNorm):
52
  def __init__(self, normalized_shape: torch.Size, bias: bool = True, **kwargs):
53
  r"""
54
- use_cache (`bool`, default = True): whether to use the bias term.
55
  """
56
  super().__init__(normalized_shape, **kwargs)
57
  if not bias:
 
51
  class LayerNorm(nn.LayerNorm):
52
  def __init__(self, normalized_shape: torch.Size, bias: bool = True, **kwargs):
53
  r"""
54
+ bias (`bool`, default = True): whether to use the bias term.
55
  """
56
  super().__init__(normalized_shape, **kwargs)
57
  if not bias: