Update custom_decoder.py

qwertyuu · Oct 14, 2021 · be829bd · be829bd
1 parent bbc6e3c
commit be829bd
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tacotron/models/custom_decoder.py b/tacotron/models/custom_decoder.py
@@ -44,8 +44,6 @@ def __init__(self, cell, helper, initial_state, output_layer=None):
 		Raises:
 			TypeError: if `cell`, `helper` or `output_layer` have an incorrect type.
 		"""
-		if not rnn_cell_impl._like_rnncell(cell):  # pylint: disable=protected-access
-			raise TypeError("cell must be an RNNCell, received: %s" % type(cell))
 		if not isinstance(helper, helper_py.Helper):
 			raise TypeError("helper must be a Helper, received: %s" % type(helper))
 		if (output_layer is not None
@@ -136,4 +134,4 @@ def step(self, time, inputs, state, name=None):
 					stop_token_prediction=stop_token)
 
 		outputs = CustomDecoderOutput(cell_outputs, stop_token, sample_ids)
-		return (outputs, next_state, next_inputs, finished)
+		return (outputs, next_state, next_inputs, finished)