def load(self, fp: str, title: str = "OUT"):
        """
        Try to load all components from specified path under a folder with given title
        """

        path = os.path.join(fp, title)

        components = self.get_components()
        for name in components:
            try:
                components[name] = torch.load(
                    os.path.join(path, name) + ".pt", map_location="cpu"
                )
            except:
                print(f"Failed to load component: {name}, continuing.")

        Try to load all components from specified path under a folder with given title
        
promptist/trlx/trlx/model/__init__.py