retutn patcher to calc model size (#34)

This commit is contained in:
Chengsong Zhang
2024-02-03 21:22:20 -06:00
committed by GitHub
parent bafb2bce54
commit 8c9dc0d0dc

View File

@@ -58,7 +58,7 @@ class UnetPatcher(ModelPatcher):
patcher = ModelPatcher(model=m, load_device=self.load_device, offload_device=self.offload_device)
self.add_extra_model_patcher_during_sampling(patcher)
return
return patcher
def add_patched_controlnet(self, cnet):
cnet.set_previous_controlnet(self.controlnet_linked_list)