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()