def setup(self) -> None:
        """Load the model into memory to make running multiple predictions efficient"""

        self.ip_adapter_dir = "models/IP_Adapter/h94/IP-Adapter/models"  # cached h94/IP-Adapter

        self.inference_config = OmegaConf.load("example/config/base.yaml")
        self.stable_diffusion_dir = self.inference_config.pretrained_model_path
        self.pia_path = self.inference_config.generate.model_path
        self.style_configs = {k: OmegaConf.load(v) for k, v in STYLE_CONFIG_LIST.items()}
        self.pipeline_dict = self.load_model_list()
Load the model into memory to make running multiple predictions efficient
predict.py