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

i

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 16:48:17 -08:00
parent f2ac9ed6e3
commit 7e0cd2ed86
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_samplers_common.py
+1 -1
View File
@@ -186,7 +186,7 @@ def apply_refiner(cfg_denoiser, x):
cfg_denoiser.p.setup_conds() cfg_denoiser.p.setup_conds()
cfg_denoiser.update_inner_model() cfg_denoiser.update_inner_model()
inference_memory = 0 inference_memory = refiner.current_controlnet_required_memory
unet_patcher = refiner.forge_objects.unet unet_patcher = refiner.forge_objects.unet
model_management.load_models_gpu( model_management.load_models_gpu(
[unet_patcher], [unet_patcher],
modules/sd_samplers_kdiffusion.py
+2 -2
View File
@@ -141,7 +141,7 @@ class KDiffusionSampler(sd_samplers_common.Sampler):
return sigmas return sigmas
def sample_img2img(self, p, x, noise, conditioning, unconditional_conditioning, steps=None, image_conditioning=None): def sample_img2img(self, p, x, noise, conditioning, unconditional_conditioning, steps=None, image_conditioning=None):
inference_memory = 0 inference_memory = self.model_wrap.inner_model.current_controlnet_required_memory
unet_patcher = self.model_wrap.inner_model.forge_objects.unet unet_patcher = self.model_wrap.inner_model.forge_objects.unet
ldm_patched.modules.model_management.load_models_gpu( ldm_patched.modules.model_management.load_models_gpu(
[unet_patcher], [unet_patcher],
@@ -205,7 +205,7 @@ class KDiffusionSampler(sd_samplers_common.Sampler):
return samples return samples
def sample(self, p, x, conditioning, unconditional_conditioning, steps=None, image_conditioning=None): def sample(self, p, x, conditioning, unconditional_conditioning, steps=None, image_conditioning=None):
inference_memory = 0 inference_memory = self.model_wrap.inner_model.current_controlnet_required_memory
unet_patcher = self.model_wrap.inner_model.forge_objects.unet unet_patcher = self.model_wrap.inner_model.forge_objects.unet
ldm_patched.modules.model_management.load_models_gpu( ldm_patched.modules.model_management.load_models_gpu(
[unet_patcher], [unet_patcher],
modules/sd_samplers_timesteps.py
+2 -2
View File
@@ -98,7 +98,7 @@ class CompVisSampler(sd_samplers_common.Sampler):
return timesteps return timesteps
def sample_img2img(self, p, x, noise, conditioning, unconditional_conditioning, steps=None, image_conditioning=None): def sample_img2img(self, p, x, noise, conditioning, unconditional_conditioning, steps=None, image_conditioning=None):
inference_memory = 0 inference_memory = self.model_wrap.inner_model.current_controlnet_required_memory
unet_patcher = self.model_wrap.inner_model.forge_objects.unet unet_patcher = self.model_wrap.inner_model.forge_objects.unet
ldm_patched.modules.model_management.load_models_gpu( ldm_patched.modules.model_management.load_models_gpu(
[unet_patcher], [unet_patcher],
@@ -150,7 +150,7 @@ class CompVisSampler(sd_samplers_common.Sampler):
return samples return samples
def sample(self, p, x, conditioning, unconditional_conditioning, steps=None, image_conditioning=None): def sample(self, p, x, conditioning, unconditional_conditioning, steps=None, image_conditioning=None):
inference_memory = 0 inference_memory = self.model_wrap.inner_model.current_controlnet_required_memory
unet_patcher = self.model_wrap.inner_model.forge_objects.unet unet_patcher = self.model_wrap.inner_model.forge_objects.unet
ldm_patched.modules.model_management.load_models_gpu( ldm_patched.modules.model_management.load_models_gpu(
[unet_patcher], [unet_patcher],