Skip to content

GitLab

"comfy/ldm/vscode:/vscode.git/clone" did not exist on "d0165d819afe76bd4e6bdd710eb5f3e571b6a804"
Commit a5056cfb authored by comfyanonymous's avatar comfyanonymous
Browse files

Remove useless code.

parent b12b48e1
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 6 deletions
comfy/ldm/modules/attention.py
View file @ a5056cfb
......@@ -104,9 +104,7 @@ def attention_basic(q, k, v, heads, mask=None):
# force cast to fp32 to avoid overflowing
if _ATTN_PRECISION =="fp32":
with torch.autocast(enabled=False, device_type = 'cuda'):
q, k = q.float(), k.float()
sim = einsum('b i d, b j d -> b i j', q, k) * scale
sim = einsum('b i d, b j d -> b i j', q.float(), k.float()) * scale
else:
sim = einsum('b i d, b j d -> b i j', q, k) * scale
......
comfy_extras/nodes_sag.py
View file @ a5056cfb
......@@ -27,9 +27,7 @@ def attention_basic_with_sim(q, k, v, heads, mask=None):
# force cast to fp32 to avoid overflowing
if _ATTN_PRECISION =="fp32":
with torch.autocast(enabled=False, device_type = 'cuda'):
q, k = q.float(), k.float()
sim = einsum('b i d, b j d -> b i j', q, k) * scale
sim = einsum('b i d, b j d -> b i j', q.float(), k.float()) * scale
else:
sim = einsum('b i d, b j d -> b i j', q, k) * scale
......
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