From fb4cd18a7fb04fe8bdba73864fb9c1341421bc0d Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Sun, 31 Mar 2024 01:13:49 -0500 Subject: [PATCH] Only expose reference_attn for now since it's the only one with code support (adain coming soon) --- adv_control/control_reference.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/adv_control/control_reference.py b/adv_control/control_reference.py index 895d3cb..c24932f 100644 --- a/adv_control/control_reference.py +++ b/adv_control/control_reference.py @@ -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: