def save(self, path, layer):
        """
        Saves centroids and assignments.

        Args:
            - path: folder used to save centroids and assignments
        """

        torch.save(self.centroids, os.path.join(path, "{}_centroids.pth".format(layer)))
        torch.save(
            self.assignments, os.path.join(path, "{}_assignments.pth".format(layer))
        )
        torch.save(self.objective, os.path.join(path, "{}_objective.pth".format(layer)))

        Saves centroids and assignments.

        Args:
            - path: folder used to save centroids and assignments
        
structured_prompting/fairseq-version/fairseq/fairseq/modules/quantization/pq/em.py