Start Align from HF + pass compute through align (#57)

* start alignment from hf model and pass target compute and capacity through * bump version
arcee-ai · Jul 15, 2024 · 9879f28 · 9879f28
1 parent bf866a9
commit 9879f28
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git 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
@@ -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)