def io_config(self) -> dict[str, Any]:
        """Return io config of the model."""
        if not self._io_config:
            return None

        io_config = self._io_config
        if isinstance(io_config, (str, Callable)):
            user_module_loader = UserModuleLoader(self.model_script, self.script_dir)
            io_config = user_module_loader.call_object(io_config, self)

        return self.get_resolved_io_config(io_config, model_attributes=self.model_attributes)
Return io config of the model.
olive/model/handler/pytorch.py