fred-dev/comfy_ui_ali

1import pickle2 3load = pickle.load4 5class Empty:6    pass7 8class Unpickler(pickle.Unpickler):9    def find_class(self, module, name):10        #TODO: safe unpickle11        if module.startswith("pytorch_lightning"):12            return Empty13        return super().find_class(module, name)14