(
self,
seperate_cfg: bool = False,
take_over: bool = False,
input_params: tuple[str] = None,
input_params_posi: dict[str, str] = None,
input_params_nega: dict[str, str] = None,
onload_model_names: tuple[str] = None
)
| 156 | |
| 157 | class PipelineUnit: |
| 158 | def __init__( |
| 159 | self, |
| 160 | seperate_cfg: bool = False, |
| 161 | take_over: bool = False, |
| 162 | input_params: tuple[str] = None, |
| 163 | input_params_posi: dict[str, str] = None, |
| 164 | input_params_nega: dict[str, str] = None, |
| 165 | onload_model_names: tuple[str] = None |
| 166 | ): |
| 167 | self.seperate_cfg = seperate_cfg |
| 168 | self.take_over = take_over |
| 169 | self.input_params = input_params |
| 170 | self.input_params_posi = input_params_posi |
| 171 | self.input_params_nega = input_params_nega |
| 172 | self.onload_model_names = onload_model_names |
| 173 | |
| 174 | |
| 175 | def process(self, pipe: BasePipeline, inputs: dict, positive=True, **kwargs) -> dict: |
nothing calls this directly
no outgoing calls
no test coverage detected