1
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00

Remove pytorch_lightning dependency.

This commit is contained in:
comfyanonymous
2023-06-13 10:11:33 -04:00
parent cb180b9998
commit 735ac4cf81
3 changed files with 15 additions and 2 deletions

View File

@@ -0,0 +1,13 @@
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)