diff --git a/prompt_control/attention_couple_ppm.py b/prompt_control/attention_couple_ppm.py index c3aaeeb..706d556 100644 --- a/prompt_control/attention_couple_ppm.py +++ b/prompt_control/attention_couple_ppm.py @@ -52,7 +52,7 @@ class Proxy: return self def __call__(self, *args, **kwargs): - return self.function(*args, *kwargs) + return self.function(*args, **kwargs) class AttentionCoupleHook(TransformerOptionsHook):