mirror of
https://github.com/Stability-AI/StableSwarmUI.git
synced 2026-10-12 02:09:42 +00:00
14 lines
367 B
Python
14 lines
367 B
Python
# This entire file is copied from ComfyUI and is just a micro-hack to avoid a dependency on PyTorch Lightning
|
|
import pickle
|
|
|
|
load = pickle.load
|
|
|
|
class Empty:
|
|
pass
|
|
|
|
class Unpickler(pickle.Unpickler):
|
|
def find_class(self, module, name):
|
|
if module.startswith("pytorch_lightning"):
|
|
return Empty
|
|
return super().find_class(module, name)
|