                        vram_total=int(stats["total"]))
          logger.debug(retval)
          return retval
  
      def _set_parallel_processing(self, multiprocess):
          """ Set whether to run detect, align, and mask together or separately.
  
          Parameters
          ----------
          multiprocess: bool
 
</s> remove         retval = dict(count=gpu_stats.device_count,
                      device=stats["device"],
                      vram_free=int(stats["free"] - vram_buffer),
                      vram_total=int(stats["total"]))
 </s> add         retval: Dict[str, Union[int, str]] = dict(count=gpu_stats.device_count,
                                                  device=stats["device"],
                                                  vram_free=int(stats["free"] - vram_buffer),
                                                  vram_total=int(stats["total"])) </s> remove     def _set_phases(self, multiprocess):
 </s> add     def _set_phases(self, multiprocess: bool) -> List[List[str]]: </s> remove     def get_aligner(name, disable_logging=False):
 </s> add     def get_aligner(name: str, disable_logging: bool = False) -> Type["Aligner"]: </s> remove     def get_masker(name, disable_logging=False):
 </s> add     def get_masker(name: str, disable_logging: bool = False) -> Type["Masker"]: </s> remove     def get_trainer(name, disable_logging=False):
 </s> add     def get_trainer(name: str, disable_logging: bool = False) -> "TrainerBase":
    def _set_parallel_processing(self, multiprocess: bool) -> bool:
                        vram_total=int(stats["total"]))
          logger.debug(retval)
          return retval
  
    def _set_parallel_processing(self, multiprocess: bool) -> bool:
          """ Set whether to run detect, align, and mask together or separately.
  
          Parameters
          ----------
          multiprocess: bool
keep keep keep keep replace keep keep keep keep keep
 extract: Add batch processing mode 
plugins/extract/pipeline.py