          The batch size for this iterator. Images will be returned in :class:`numpy.ndarray`
          objects of this size from the iterator.
      """
      def __init__(self,
                   config: ConfigType,
                   model: "ModelBase",
                   side: Literal["a", "b"],
                   images: List[str],
                   batch_size: int) -> None:
          logger.debug("Initializing %s: (model: %s, side: %s, images: %s , "  # type: ignore
 
</s> remove                  config: ConfigType,
 </s> add                  config: Dict[str, "ConfigValueType"], </s> remove     config: :class:`lib.config.FaceswapConfig`
 </s> add     config: dict </s> remove                  config: ConfigType) -> None:
 </s> add                  config: Dict[str, "ConfigValueType"]) -> None: </s> remove     def __init__(self, model: "ModelBase", coverage_ratio: float) -> None:
        logger.debug("Initializing %s: model: '%s', coverage_ratio: %s)",
                     self.__class__.__name__, model, coverage_ratio)
 </s> add     def __init__(self,
                 model: "ModelBase",
                 coverage_ratio: float,
                 mask_opacity: int,
                 mask_color: str) -> None:
        logger.debug("Initializing %s: model: '%s', coverage_ratio: %s, mask_opacity: %s, "
                     "mask_color: %s)",
                     self.__class__.__name__, model, coverage_ratio, mask_opacity, mask_color)
                 config: Dict[str, "ConfigValueType"],
          The batch size for this iterator. Images will be returned in :class:`numpy.ndarray`
          objects of this size from the iterator.
      """
      def __init__(self,
                 config: Dict[str, "ConfigValueType"],
                   model: "ModelBase",
                   side: Literal["a", "b"],
                   images: List[str],
                   batch_size: int) -> None:
          logger.debug("Initializing %s: (model: %s, side: %s, images: %s , "  # type: ignore
keep keep keep keep replace keep keep keep keep keep
 Preview - Allow mask color/opacity configuration
  - typing: lib.config
  - docs: lib.config
  - bugfix: popup_configure - keep assets on top 
lib/training/generator.py