Fix context creation

adapter-hub · Jan 21, 2022 · 6be596d · 6be596d
1 parent 245f051
commit 6be596d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/adapters/context.py b/src/transformers/adapters/context.py
@@ -85,7 +85,7 @@ def wrap(cls, f):
 
         @functools.wraps(f)
         def wrapper_func(self, *args, **kwargs):
-            context = cls()
+            context = cls(self)
             cls.get_contexts().append(context)
             results = f(self, *args, **kwargs)
             cls.get_contexts().pop()