cxllm-vendors/stable-diffusion-webui-forge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

retutn patcher to calc model size (#34)

Browse Source
This commit is contained in:
Chengsong Zhang
2024-02-03 21:22:20 -06:00
committed by GitHub
parent bafb2bce54
commit 8c9dc0d0dc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules_forge/unet_patcher.py
+1 -1
View File
@@ -58,7 +58,7 @@ class UnetPatcher(ModelPatcher):
patcher = ModelPatcher(model=m, load_device=self.load_device, offload_device=self.offload_device) patcher = ModelPatcher(model=m, load_device=self.load_device, offload_device=self.offload_device)
self.add_extra_model_patcher_during_sampling(patcher) self.add_extra_model_patcher_during_sampling(patcher)
return return patcher
def add_patched_controlnet(self, cnet): def add_patched_controlnet(self, cnet):
cnet.set_previous_controlnet(self.controlnet_linked_list) cnet.set_previous_controlnet(self.controlnet_linked_list)