full_or_peft option for alignment procedure (#62)

* full_or_peft option for alignment * formatting --------- Co-authored-by: Shahrzad <[email protected]> Co-authored-by: Jacobsolawetz <[email protected]>
arcee-ai · Jul 22, 2024 · 0101469 · 0101469
1 parent 1381deb
commit 0101469
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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,