lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-22 15:53:58 +00:00
Code Issues Packages Projects Releases Wiki 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
Download Patch File Download Diff File Expand all files Collapse all files

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
)