adding peft_config field

deepanker13 · Dec 12, 2023 · f87e8e9 · f87e8e9
1 parent a017b21
commit f87e8e9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sdk/python/kubeflow/storage_init_container/hugging_face.py b/sdk/python/kubeflow/storage_init_container/hugging_face.py
@@ -34,6 +34,7 @@ def __post_init__(self):
 @dataclass
 class HuggingFaceTrainParams:
     additional_data: Dict[str, Any] = field(default_factory=dict)
+    peft_config: Dict[str, Any] = field(default_factory=dict)
 
 
 class HuggingFace(modelProvider):