lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-26 19:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update forge_stylealign.py

Browse Source
This commit is contained in:
layerdiffusion
2024-08-03 14:49:40 -07:00
parent b3a6eeefd6
commit a17abbc097
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extensions-builtin/sd_forge_stylealign/scripts/forge_stylealign.py
View File

@@ -2,11 +2,11 @@ import torch
import gradio as gr
from modules import scripts
import ldm_patched.ldm.modules.attention as attention
from backend import attention
def sdp(q, k, v, transformer_options):
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)
class StyleAlignForForge(scripts.Script):