mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-17 00:23:14 +00:00
Brandon Thomas
cd06f71dc7
test commit remove submodule Add ComfyUI @ c5a369a33ddb622827552716d9b0119035a2e666 Comfy upstream SHA: c5a369a33ddb622827552716d9b0119035a2e666 add custom node dependnencies add large checkpoint for comfyui
14 lines
286 B
Python
14 lines
286 B
Python
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)
|