Only expose reference_attn for now since it's the only one with code support (adain coming soon)

This commit is contained in:
Jedrzej Kosinski
2024-03-31 01:13:49 -05:00
parent e5841d1759
commit fb4cd18a7f
+2 -1
View File
@@ -124,7 +124,8 @@ class ReferenceType:
ATTN_ADAIN = "reference_attn+adain"
STYLE_ALIGN = "StyleAlign"
_LIST = [ATTN, ADAIN, ATTN_ADAIN]
_LIST = [ATTN]
_LIST_FULL = [ATTN, ADAIN, ATTN_ADAIN]
class ReferenceOptions: