From 96b95609bea7cfcd789a2c6eb0c94d139f984aaa Mon Sep 17 00:00:00 2001 From: Artem Chumachenko Date: Tue, 5 Mar 2024 13:45:41 +0400 Subject: [PATCH] style --- src/petals/utils/peft.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/petals/utils/peft.py b/src/petals/utils/peft.py index a47bf79..04b2a4a 100644 --- a/src/petals/utils/peft.py +++ b/src/petals/utils/peft.py @@ -171,6 +171,7 @@ using_adapter = AdapterContextMixin.using_adapter class LoraLinear(AdapterContextMixin, lora.Linear): """LoRA linear layer that uses adapter selected via using_adapter""" + def __init__(self, base_layer, adapter_name: str): nn.Module.__init__(self) lora.LoraLayer.__init__(self, base_layer)