add dora condition for �dd_adapter

adapter-hub · Feb 4, 2025 · 7a68d71 · 7a68d71
1 parent d6d1dab
commit 7a68d71
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/adapters/methods/lora.py b/src/adapters/methods/lora.py
@@ -280,9 +280,11 @@ def add_adapter(self, adapter_name: str, layer_idx: int) -> bool:
         )
         if lora_config is not None and self._check_lora_location(lora_config):
             if lora_config.composition_mode == "add":
-                lora_cls = DoRA
+                lora_cls = LoRA
             elif lora_config.composition_mode == "scale":
                 lora_cls = IA3
+            elif lora_config.composition_mode == "dora":
+                lora_cls = DoRA
             else:
                 raise ValueError(f"Unknown composition_mode: {lora_config.composition_mode}")
             lora = lora_cls(