Spaces:
Runtime error
Runtime error
File size: 1,175 Bytes
2e82449 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
from modules_forge.shared import add_supported_control_model
from modules_forge.supported_controlnet import ControlModelPatcher
from lib_controllllite.lib_controllllite import LLLiteLoader
opLLLiteLoader = LLLiteLoader().load_lllite
class ControlLLLitePatcher(ControlModelPatcher):
@staticmethod
def try_build_from_state_dict(state_dict, ckpt_path):
if not any('lllite' in k for k in state_dict.keys()):
return None
return ControlLLLitePatcher(state_dict)
def __init__(self, state_dict):
super().__init__()
self.state_dict = state_dict
return
def process_before_every_sampling(self, process, cond, mask, *args, **kwargs):
unet = process.sd_model.forge_objects.unet
unet = opLLLiteLoader(
model=unet,
state_dict=self.state_dict,
cond_image=cond.movedim(1, -1),
strength=self.strength,
steps=process.steps,
start_percent=self.start_percent,
end_percent=self.end_percent
)[0]
process.sd_model.forge_objects.unet = unet
return
add_supported_control_model(ControlLLLitePatcher)
|