diff --git a/arcee/__init__.py b/arcee/__init__.py index 4ebb7dc..e689649 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 ee49530..1758ddc 100644 --- a/arcee/api.py +++ b/arcee/api.py @@ -353,6 +353,7 @@ def start_alignment( pretrained_model: Optional[str] = None, merging_model: Optional[str] = None, alignment_model: Optional[str] = None, + hf_model: Optional[str] = None, target_compute: Optional[str] = None, capacity_id: Optional[str] = None, ) -> Dict[str, str]: @@ -377,14 +378,11 @@ def start_alignment( "pretrained_model": pretrained_model, "merging_model": merging_model, "alignment_model": alignment_model, + "hf_model": hf_model, + "target_compute": target_compute, + "capacity_id": capacity_id, } - if target_compute: - data["target_compute"] = target_compute - - if capacity_id: - data["capacity_id"] = capacity_id - # Assuming make_request is a function that handles the request, it's called here return make_request("post", Route.alignment + "/startAlignment", data)