fix: Add name and description classmethods to IA3Config (#3844)

ludwig-ai · Dec 20, 2023 · e1edcbc · e1edcbc
1 parent 8f9b546
commit e1edcbc
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/ludwig/schema/llms/peft.py b/ludwig/schema/llms/peft.py
@@ -524,6 +524,14 @@ def to_config(self, task_type: str = None, **kwargs) -> "PeftConfig":
             task_type=task_type,
         )
 
+    @classmethod
+    def name(cls) -> str:
+        return "IA3"
+
+    @classmethod
+    def description(cls) -> str:
+        return LLM_METADATA["adapter"]["ia3"]["type"].long_description
+
 
 @DeveloperAPI
 def get_adapter_conds():