mirror of
https://github.com/Stability-AI/stablediffusion.git
synced 2024-12-22 15:44:58 +00:00
change type to attn_type, Fixing bug which doesnot have the variable type. Authors have failed to change it to type to attn_type. Curious if cross attn is not used in stable diffusion?
This commit is contained in:
parent
cf1d67a6fd
commit
8b6e99fbdd
1 changed files with 1 additions and 1 deletions
|
@ -288,7 +288,7 @@ def make_attn(in_channels, attn_type="vanilla", attn_kwargs=None):
|
||||||
elif attn_type == "vanilla-xformers":
|
elif attn_type == "vanilla-xformers":
|
||||||
print(f"building MemoryEfficientAttnBlock with {in_channels} in_channels...")
|
print(f"building MemoryEfficientAttnBlock with {in_channels} in_channels...")
|
||||||
return MemoryEfficientAttnBlock(in_channels)
|
return MemoryEfficientAttnBlock(in_channels)
|
||||||
elif type == "memory-efficient-cross-attn":
|
elif attn_type == "memory-efficient-cross-attn":
|
||||||
attn_kwargs["query_dim"] = in_channels
|
attn_kwargs["query_dim"] = in_channels
|
||||||
return MemoryEfficientCrossAttentionWrapper(**attn_kwargs)
|
return MemoryEfficientCrossAttentionWrapper(**attn_kwargs)
|
||||||
elif attn_type == "none":
|
elif attn_type == "none":
|
||||||
|
|
Loading…
Reference in a new issue