Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

working with 1.4, 1.5, not with inpainting 1.5

Browse Source
This commit is contained in:
Lincoln Stein
2022-10-26 18:25:48 -04:00
parent 9b7159720f
commit 2daf187bdb
3 changed files with 28 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
ldm/models/diffusion/ksampler.py
View File

@ -23,10 +23,9 @@ def cfg_apply_threshold(result, threshold = 0.0, scale = 0.7):
class CFGDenoiser(nn.Module):
def __init__(self, sampler, threshold = 0, warmup = 0):
def __init__(self, model, threshold = 0, warmup = 0):
super().__init__()
self.inner_model = sampler.model
self.sampler = sampler
self.inner_model = model
self.threshold = threshold
self.warmup_max = warmup
self.warmup = max(warmup / 10, 1)