list -> set

adapter-hub · Feb 10, 2025 · 8f17f5e · 8f17f5e
1 parent 096297a
commit 8f17f5e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/adapters/context.py b/src/adapters/context.py
@@ -79,15 +79,15 @@ class ForwardContext(ContextManager):
     # thread-local storage that holds a stack of active contexts
     storage = threading.local()
 
-    context_args = [
+    context_args = {
         "output_adapter_gating_scores",
         "output_adapter_fusion_attentions",
         "adapter_input_parallelized",
-    ]
-    context_attributes = [
+    }
+    context_attributes = {
         "adapter_gating_scores",
         "adapter_fusion_attentions",
-    ]
+    }
     # Additional used attributes not exposed to the user
     # - prompt_tokens_length: length of the prompt tokens