Unverified Commit d9dc993f authored by Minseo Kang's avatar Minseo Kang Committed by GitHub
Browse files

Fix typo in T5Block error message (#29881)

parent a25037be
...@@ -552,7 +552,7 @@ class MT5Block(nn.Module): ...@@ -552,7 +552,7 @@ class MT5Block(nn.Module):
if len(past_key_value) != expected_num_past_key_values: if len(past_key_value) != expected_num_past_key_values:
raise ValueError( raise ValueError(
f"There should be {expected_num_past_key_values} past states. " f"There should be {expected_num_past_key_values} past states. "
f"{'2 (past / key) for cross attention. ' if expected_num_past_key_values == 4 else ''}" f"{'2 (key / value) for cross attention. ' if expected_num_past_key_values == 4 else ''}"
f"Got {len(past_key_value)} past key / value states" f"Got {len(past_key_value)} past key / value states"
) )
......
...@@ -587,7 +587,7 @@ class Pop2PianoBlock(nn.Module): ...@@ -587,7 +587,7 @@ class Pop2PianoBlock(nn.Module):
if len(past_key_value) != expected_num_past_key_values: if len(past_key_value) != expected_num_past_key_values:
raise ValueError( raise ValueError(
f"There should be {expected_num_past_key_values} past states. " f"There should be {expected_num_past_key_values} past states. "
f"{'2 (past / key) for cross attention. ' if expected_num_past_key_values == 4 else ''}" f"{'2 (key / value) for cross attention. ' if expected_num_past_key_values == 4 else ''}"
f"Got {len(past_key_value)} past key / value states" f"Got {len(past_key_value)} past key / value states"
) )
......
...@@ -677,7 +677,7 @@ class T5Block(nn.Module): ...@@ -677,7 +677,7 @@ class T5Block(nn.Module):
if len(past_key_value) != expected_num_past_key_values: if len(past_key_value) != expected_num_past_key_values:
raise ValueError( raise ValueError(
f"There should be {expected_num_past_key_values} past states. " f"There should be {expected_num_past_key_values} past states. "
f"{'2 (past / key) for cross attention. ' if expected_num_past_key_values == 4 else ''}" f"{'2 (key / value) for cross attention. ' if expected_num_past_key_values == 4 else ''}"
f"Got {len(past_key_value)} past key / value states" f"Got {len(past_key_value)} past key / value states"
) )
......
...@@ -624,7 +624,7 @@ class TFT5Block(keras.layers.Layer): ...@@ -624,7 +624,7 @@ class TFT5Block(keras.layers.Layer):
if len(past_key_value) != expected_num_past_key_values: if len(past_key_value) != expected_num_past_key_values:
raise ValueError( raise ValueError(
f"There should be {expected_num_past_key_values} past states. " f"There should be {expected_num_past_key_values} past states. "
f"{'2 (past / key) for cross attention' if expected_num_past_key_values == 4 else ''}. " f"{'2 (key / value) for cross attention' if expected_num_past_key_values == 4 else ''}. "
f"Got {len(past_key_value)} past key / value states" f"Got {len(past_key_value)} past key / value states"
) )
......
...@@ -940,7 +940,7 @@ class UdopBlock(nn.Module): ...@@ -940,7 +940,7 @@ class UdopBlock(nn.Module):
if len(past_key_value) != expected_num_past_key_values: if len(past_key_value) != expected_num_past_key_values:
raise ValueError( raise ValueError(
f"There should be {expected_num_past_key_values} past states. " f"There should be {expected_num_past_key_values} past states. "
f"{'2 (past / key) for cross attention. ' if expected_num_past_key_values == 4 else ''}" f"{'2 (key / value) for cross attention. ' if expected_num_past_key_values == 4 else ''}"
f"Got {len(past_key_value)} past key / value states" f"Got {len(past_key_value)} past key / value states"
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment