| # Taken from https://github.com/comfyanonymous/ComfyUI | |
| # This file is only for reference, and not used in the backend or runtime. | |
| import pickle | |
| load = pickle.load | |
| class Empty: | |
| pass | |
| class Unpickler(pickle.Unpickler): | |
| def find_class(self, module, name): | |
| #TODO: safe unpickle | |
| if module.startswith("pytorch_lightning"): | |
| return Empty | |
| return super().find_class(module, name) | |