Jonathan Tow
commited on
Commit
•
1465d75
1
Parent(s):
eee47ee
fix: update `LayerNorm` docstring
Browse files
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 |
-
|
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:
|