From 4a18bed09594f1a0fcd667bc96d3636745c51493 Mon Sep 17 00:00:00 2001 From: Shahrzad Date: Tue, 9 Jul 2024 10:29:12 -0600 Subject: [PATCH 1/2] full_or_peft option for alignment --- arcee/__init__.py | 2 +- arcee/api.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/arcee/__init__.py b/arcee/__init__.py index c5b99fd..694aa67 100644 --- a/arcee/__init__.py +++ b/arcee/__init__.py @@ -1,4 +1,4 @@ -__version__ = "1.3.2" +__version__ = "1.3.3" import os diff --git a/arcee/api.py b/arcee/api.py index a523b75..19f4148 100644 --- a/arcee/api.py +++ b/arcee/api.py @@ -342,6 +342,7 @@ def start_alignment( alignment_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. @@ -364,6 +365,7 @@ def start_alignment( "pretrained_model": pretrained_model, "merging_model": merging_model, "alignment_model": alignment_model, + "full_or_peft": full_or_peft, } if target_compute: From 2b109ac633ce2ca9c52231c2edaff9f0a3e92b35 Mon Sep 17 00:00:00 2001 From: Shahrzad Date: Tue, 9 Jul 2024 10:30:15 -0600 Subject: [PATCH 2/2] formatting --- arcee/api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/arcee/api.py b/arcee/api.py index 19f4148..f15a4eb 100644 --- a/arcee/api.py +++ b/arcee/api.py @@ -342,7 +342,7 @@ def start_alignment( alignment_model: Optional[str] = None, target_compute: Optional[str] = None, capacity_id: Optional[str] = None, - full_or_peft: Optional[str] = "full" + full_or_peft: Optional[str] = "full", ) -> Dict[str, str]: """ Start the alignment of a model.