Spaces:
Running
on
A10G
Running
on
A10G
Fix typo in lm.py
#22
by
eltociear
- opened
- audiocraft/models/lm.py +1 -1
audiocraft/models/lm.py
CHANGED
@@ -183,7 +183,7 @@ class LMModel(StreamingModule):
|
|
183 |
depthwise_init (Optional[str]): Depwthwise initialization strategy. The following options are valid:
|
184 |
'current' where the depth corresponds to the current layer index or 'global' where the total number
|
185 |
of layer is used as depth. If not set, no depthwise initialization strategy is used.
|
186 |
-
zero_bias_init (bool): Whether to
|
187 |
"""
|
188 |
assert depthwise_init is None or depthwise_init in ['current', 'global']
|
189 |
assert depthwise_init is None or weight_init is not None, \
|
|
|
183 |
depthwise_init (Optional[str]): Depwthwise initialization strategy. The following options are valid:
|
184 |
'current' where the depth corresponds to the current layer index or 'global' where the total number
|
185 |
of layer is used as depth. If not set, no depthwise initialization strategy is used.
|
186 |
+
zero_bias_init (bool): Whether to initialize bias to zero or not.
|
187 |
"""
|
188 |
assert depthwise_init is None or depthwise_init in ['current', 'global']
|
189 |
assert depthwise_init is None or weight_init is not None, \
|