code: def ae_kl_f16(
      size: int = 256, pretrained: bool = True, **kwargs: Any
  ) -> TLoadZoo:
      if pretrained and size != 256:
          raise ValueError(
              "pretrained `ae_kl_f16` should have `size`=256"
          )
      kwargs["img_size"] = size
      kwargs["pretrained"] = pretrained
      return _load("ae/kl.f16", **kwargs)
  
imports: from typing import Any
  from typing import Union
  from cftool.types import tensor_dict_type
  from ...schema import DLConfig
  from ...zoo.core import DLZoo
  from ...zoo.core import TPipeline
  from ...models.cv.diffusion.utils import CONCAT_TYPE
label_desc: Use image node_id 4 for calling a global function with example usage: ae_kl_f16(size, pretrained) and returns: _load