@@ -147,10 +147,9 @@ def __init__(
147147 name : str | None = None ,
148148 ) -> None :
149149 super ().__init__ ()
150- assert embed_dim > 0 , (
151- "Expected embed_dim to be greater than 0, "
152- f"but received { embed_dim } "
153- )
150+ assert (
151+ embed_dim > 0
152+ ), f"Expected embed_dim to be greater than 0, but received { embed_dim } "
154153 self ._dtype = self ._helper .get_default_dtype ()
155154 self ._bias_attr = bias_attr
156155 self ._weight_attr = weight_attr
@@ -338,13 +337,12 @@ def __init__(
338337 ) -> None :
339338 super ().__init__ ()
340339
341- assert embed_dim > 0 , (
342- "Expected embed_dim to be greater than 0, "
343- f"but received { embed_dim } "
344- )
345- assert num_heads > 0 , (
346- "Expected nhead to be greater than 0, " f"but received { num_heads } "
347- )
340+ assert (
341+ embed_dim > 0
342+ ), f"Expected embed_dim to be greater than 0, but received { embed_dim } "
343+ assert (
344+ num_heads > 0
345+ ), f"Expected nhead to be greater than 0, but received { num_heads } "
348346
349347 self .normalize_before = normalize_before
350348 self ._dtype = self ._helper .get_default_dtype ()
@@ -830,12 +828,12 @@ def __init__(
830828 self ._config .pop ("__class__" , None ) # py3
831829
832830 super ().__init__ ()
833- assert d_model > 0 , (
834- "Expected d_model to be greater than 0, " f"but received { d_model } "
835- )
836- assert nhead > 0 , (
837- "Expected nhead to be greater than 0, " f"but received { nhead } "
838- )
831+ assert (
832+ d_model > 0
833+ ), f"Expected d_model to be greater than 0, but received { d_model } "
834+ assert (
835+ nhead > 0
836+ ), f"Expected nhead to be greater than 0, but received { nhead } "
839837 assert dim_feedforward > 0 , (
840838 "Expected dim_feedforward to be greater than 0, "
841839 f"but received { dim_feedforward } "
@@ -1306,13 +1304,12 @@ def __init__(
13061304 ) -> None :
13071305 super ().__init__ ()
13081306
1309- assert embed_dim > 0 , (
1310- "Expected embed_dim to be greater than 0, "
1311- f"but received { embed_dim } "
1312- )
1313- assert num_heads > 0 , (
1314- "Expected nhead to be greater than 0, " f"but received { num_heads } "
1315- )
1307+ assert (
1308+ embed_dim > 0
1309+ ), f"Expected embed_dim to be greater than 0, but received { embed_dim } "
1310+ assert (
1311+ num_heads > 0
1312+ ), f"Expected nhead to be greater than 0, but received { num_heads } "
13161313 assert (
13171314 dim_feedforward > 0
13181315 ), f"Expected dim_feedforward to be greater than 0, but received { dim_feedforward } "
0 commit comments