def get_dependencies(self) -> Set[ModelPart]:
deps = ModelPart.get_dependencies(self)
# feed only if needed
if isinstance(self.input_sequence, ModelPart):
feedable = cast(ModelPart, self.input_sequence)
deps |= feedable.get_dependencies()
return deps
def get_dependencies(self) -> Set[ModelPart]:
"""Collect recusively all encoders and decoders."""
deps = ModelPart.get_dependencies(self)
# feed only if needed
if isinstance(self.input_sequence, ModelPart):
feedable = cast(ModelPart, self.input_sequence)
deps = deps.union(feedable.get_dependencies())
return deps
def get_dependencies(self) -> Set[ModelPart]:
"""Collect recusively all inputs."""
to_return = ModelPart.get_dependencies(self)
if (self.input_for_cross_attention is not None
and isinstance(self.input_for_cross_attention, ModelPart)):
to_return = to_return.union(
self.input_for_cross_attention.get_dependencies())
return to_return
def __init__(self, output_series: str, decoder: ModelPart) -> None:
self.output_series = output_series
self._decoder = decoder
self.all_coders = decoder.get_dependencies()
