soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

elegant codes

Browse Source
This commit is contained in:
lllyasviel 2024-02-01 23:04:24 -08:00
parent a9212b640f
commit 24db0e241a
2 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
extensions-builtin/sd_forge_photomaker/scripts/forge_photomaker.py
View File

@ -52,14 +52,7 @@ class PhotomakerPatcher(ControlModelPatcher):
text = process.prompts[0]
cond_modified = opPhotoMakerEncode(photomaker=self.model, image=cond.movedim(1, -1), clip=clip, text=text)[0]
noise = kwargs['x']
cond_modified = encode_model_conds(
model_function=unet.model.extra_conds,
conds=convert_cond(cond_modified),
noise=noise,
device=noise.device,
prompt_type="positive"
)[0]
cond_modified = unet.encode_conds_from_clip(conds=cond_modified, noise=kwargs['x'])[0]
def conditioning_modifier(model, x, timestep, uncond, cond, cond_scale, model_options, seed):
cond = cond.copy()

11
modules_forge/unet_patcher.py
View File

@ -1,6 +1,8 @@
import copy
from ldm_patched.modules.model_patcher import ModelPatcher
from ldm_patched.modules.sample import convert_cond
from ldm_patched.modules.samplers import encode_model_conds
class UnetPatcher(ModelPatcher):
@ -96,3 +98,12 @@ class UnetPatcher(ModelPatcher):
for transformer_index in range(16):
self.set_model_patch_replace(patch, target, block_name, number, transformer_index)
return
def encode_conds_from_clip(self, conds, noise, prompt_type="positive"):
return encode_model_conds(
model_function=self.model.extra_conds,
conds=convert_cond(conds),
noise=noise,
device=noise.device,
prompt_type=prompt_type
)