Skip to content

S
Stable Diffusion Webui
Commit c84c9df7 authored by FNSpd's avatar FNSpd Committed by GitHub
Browse files
Options

Update sd_hijack_optimizations.py

parent 91cfa971
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
modules/sd_hijack_optimizations.py
View file @ c84c9df7
...@@ -337,7 +337,7 @@ def xformers_attention_forward(self, x, context=None, mask=None): ...@@ -337,7 +337,7 @@ def xformers_attention_forward(self, x, context=None, mask=None):
dtype = q.dtype dtype = q.dtype
if shared.opts.upcast_attn: if shared.opts.upcast_attn:
q, k = q.float(), k.float() q, k, v = q.float(), k.float(), v.float()
out = xformers.ops.memory_efficient_attention(q, k, v, attn_bias=None, op=get_xformers_flash_attention_op(q, k, v)) out = xformers.ops.memory_efficient_attention(q, k, v, attn_bias=None, op=get_xformers_flash_attention_op(q, k, v))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment