seekrai 0.2.0__tar.gz → 0.2.1__tar.gz

{seekrai-0.2.0 → seekrai-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: seekrai
-Version: 0.2.0
+Version: 0.2.1
 Summary: Python client for SeekrAI
 Home-page: https://gitlab.cb.ntent.com/ml/seekr-py
 License: Apache-2.0

{seekrai-0.2.0 → seekrai-0.2.1}/pyproject.toml

@@ -14,7 +14,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "seekrai"
-version = "0.2.0"
+version = "0.2.1"
 authors = [
     "SeekrFlow <support@seekr.com>"
 ]

{seekrai-0.2.0 → seekrai-0.2.1}/src/seekrai/types/finetune.py

@@ -118,7 +118,6 @@ class InfrastructureConfig(BaseModel):
     accel_type: AcceleratorType
     n_accel: int
 
-
 class FinetuneRequest(BaseModel):
     """
     Fine-tune request type
@@ -141,14 +140,16 @@ class FinetuneResponse(BaseModel):
     # validation_files: str | None = None TODO
     # base model name
     model: str | None = None
+    accel_type: AcceleratorType
+    n_accel: int
     # number of epochs
-    # n_epochs: int | None = None
+    n_epochs: int | None = None
     # number of checkpoints to save
     # n_checkpoints: int | None = None # TODO
     # training batch size
-    # batch_size: int | None = None
+    batch_size: int | None = None
     # training learning rate
-    # learning_rate: float | None = None
+    learning_rate: float | None = None
     # number of steps between evals
     # eval_steps: int | None = None TODO
     # is LoRA finetune boolean
@@ -159,6 +160,8 @@ class FinetuneResponse(BaseModel):
     # created/updated datetime stamps
     created_at: datetime | None = None
     # updated_at: str | None = None
+    # up to 40 character suffix for output model name
+    experiment_name: str | None = None
     # job status
     status: FinetuneJobStatus | None = None