Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: extract attribute controlnet_softness

Browse Source
This commit is contained in:
Manuel Schmid
2024-01-22 20:09:24 +01:00
parent 78d2ec8d77
commit f3222b0f27
6 changed files with 9 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/patch.py
+4 -4
View File
@@ -17,7 +17,6 @@ import ldm_patched.controlnet.cldm
import ldm_patched.modules.model_patcher
import ldm_patched.modules.samplers
import ldm_patched.modules.args_parser
import modules.advanced_parameters as advanced_parameters
import warnings
import safetensors.torch
import modules.constants as constants
@@ -37,11 +36,12 @@ adm_scaler_end = 0.3
positive_adm_scale = 1.5
negative_adm_scale = 0.8
controlnet_softness = 0.25
adaptive_cfg = 7.0
global_diffusion_progress = 0
eps_record = None
def calculate_weight_patched(self, patches, weight, key):
for p in patches:
alpha = p[0]
@@ -359,10 +359,10 @@ def patched_cldm_forward(self, x, hint, timesteps, context, y=None, **kwargs):
h = self.middle_block(h, emb, context)
outs.append(self.middle_block_out(h, emb, context))
if advanced_parameters.controlnet_softness > 0:
if controlnet_softness > 0:
for i in range(10):
k = 1.0 - float(i) / 9.0
outs[i] = outs[i] * (1.0 - advanced_parameters.controlnet_softness * k)
outs[i] = outs[i] * (1.0 - controlnet_softness * k)
return outs