From 0101469c2ae8edef149ea04ff4201554a821a01d Mon Sep 17 00:00:00 2001 From: shahrzads <56282669+shahrzads@users.noreply.github.com> Date: Mon, 22 Jul 2024 08:26:13 -0600 Subject: [PATCH] full_or_peft option for alignment procedure (#62) * full_or_peft option for alignment * formatting --------- Co-authored-by: Shahrzad Co-authored-by: Jacobsolawetz --- arcee/api.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/arcee/api.py b/arcee/api.py index 1758ddc..fbcac32 100644 --- a/arcee/api.py +++ b/arcee/api.py @@ -356,6 +356,7 @@ def start_alignment( hf_model: Optional[str] = None, target_compute: Optional[str] = None, capacity_id: Optional[str] = None, + full_or_peft: Optional[str] = "full", ) -> Dict[str, str]: """ Start the alignment of a model. @@ -378,6 +379,7 @@ def start_alignment( "pretrained_model": pretrained_model, "merging_model": merging_model, "alignment_model": alignment_model, + "full_or_peft": full_or_peft, "hf_model": hf_model, "target_compute": target_compute, "capacity_id": capacity_id,