lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-05 07:29:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

clipvision, ipadapter, and misc

Browse Source 
backend is 75% finished
This commit is contained in:
layerdiffusion
2024-08-03 13:37:22 -07:00
parent bb5083f3c2
commit 07b2d2ccac
9 changed files with 474 additions and 320 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py
View File

@@ -2,15 +2,15 @@ import torch
from modules_forge.supported_preprocessor import Preprocessor, PreprocessorParameter
from modules_forge.shared import add_supported_preprocessor
from ldm_patched.modules.samplers import sampling_function
import ldm_patched.ldm.modules.attention as attention
from backend.sampling.sampling_function import sampling_function_inner
from backend import attention
def sdp(q, k, v, transformer_options):
if q.shape[0] == 0:
return q
return attention.optimized_attention(q, k, v, heads=transformer_options["n_heads"], mask=None)
return attention.attention_function(q, k, v, heads=transformer_options["n_heads"], mask=None)
def adain(x, target_std, target_mean):
@@ -81,7 +81,7 @@ class PreprocessorReference(Preprocessor):
self.is_recording_style = True
xt = latent_image.to(x) + torch.randn(x.size(), dtype=x.dtype, generator=gen_cpu).to(x) * sigma
sampling_function(model, xt, timestep, uncond, cond, 1, model_options, seed)
sampling_function_inner(model, xt, timestep, uncond, cond, 1, model_options, seed)
self.is_recording_style = False