      from tensorflow.keras.models import load_model, Model as KModel  # noqa pylint:disable=import-error,no-name-in-module
  
  if TYPE_CHECKING:
      import argparse
  
  logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
  _CONFIG: Dict[str, Union[str, float, int, bool, None]] = {}
 
</s> remove     def __init__(self, config: dict) -> None:
        logger.debug("Initializing %s", self.__class__.__name__)
 </s> add     def __init__(self, config: dict, color_order: Literal["bgr", "rgb"]) -> None:
        logger.debug("Initializing %s: (color_order: %s)", self.__class__.__name__, color_order)
if sys.version_info < (3, 8):
    from typing_extensions import Literal
else:
    from typing import Literal

      from tensorflow.keras.models import load_model, Model as KModel  # noqa pylint:disable=import-error,no-name-in-module
  
if sys.version_info < (3, 8):
    from typing_extensions import Literal
else:
    from typing import Literal

  if TYPE_CHECKING:
      import argparse
  
  logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
  _CONFIG: Dict[str, Union[str, float, int, bool, None]] = {}
keep add keep keep keep keep keep
 Add Flip Loss Function
    - Add Flip for AMD and TF
    - Split Perceptual Loss functions to own modules
    - Fix allowed input shape for models
    - Allow GUI tooltip to display at higher width 
plugins/train/model/_base/model.py