Add fixes

Author: timofeev1995

src/together/cli/api/finetune.py

@@ -206,6 +206,12 @@ def fine_tuning(ctx: click.Context) -> None:
     default=None,
     help="HF API token to use to upload a checkpoint to a private repo",
 )
+@click.option(
+    "--hf-repo-to-upload",
+    type=str,
+    default=None,
+    help="HF repo to upload the fine-tuned model to",
+)
 def create(
     ctx: click.Context,
     training_file: str,
@@ -240,7 +246,8 @@ def create(
     rpo_alpha: float | None,
     simpo_gamma: float | None,
     from_checkpoint: str,
-    hf_api_token: str,
+    hf_api_token: str | None,
+    hf_repo_to_upload: str | None,
 ) -> None:
     """Start fine-tuning"""
     client: Together = ctx.obj
@@ -278,6 +285,7 @@ def create(
         simpo_gamma=simpo_gamma,
         from_checkpoint=from_checkpoint,
         hf_api_token=hf_api_token,
+        hf_repo_to_upload=hf_repo_to_upload,
     )
 
     if model is None and from_checkpoint is None:

src/together/resources/finetune.py

@@ -344,6 +344,7 @@ def create(
         simpo_gamma: float | None = None,
         from_checkpoint: str | None = None,
         hf_api_token: str | None = None,
+        hf_repo_to_upload: str | None = None,
     ) -> FinetuneResponse:
         """
         Method to initiate a fine-tuning job
@@ -401,6 +402,7 @@ def create(
                 The format: {$JOB_ID/$OUTPUT_MODEL_NAME}:{$STEP}.
                 The step value is optional, without it the final checkpoint will be used.
             hf_api_token (str, optional): API key for the Hugging Face Hub. Defaults to None.
+            hf_repo_to_upload (str, optional): HF repo to upload the fine-tuned model to. Defaults to None.
 
         Returns:
             FinetuneResponse: Object containing information about fine-tuning job.
@@ -455,6 +457,7 @@ def create(
             simpo_gamma=simpo_gamma,
             from_checkpoint=from_checkpoint,
             hf_api_token=hf_api_token,
+            hf_repo_to_upload=hf_repo_to_upload,
         )
 
         if verbose:
@@ -768,6 +771,7 @@ async def create(
         simpo_gamma: float | None = None,
         from_checkpoint: str | None = None,
         hf_api_token: str | None = None,
+        hf_repo_to_upload: str | None = None,
     ) -> FinetuneResponse:
         """
         Async method to initiate a fine-tuning job
@@ -825,6 +829,7 @@ async def create(
                 The format: {$JOB_ID/$OUTPUT_MODEL_NAME}:{$STEP}.
                 The step value is optional, without it the final checkpoint will be used.
             hf_api_token (str, optional): API key for the Huggging Face Hub. Defaults to None.
+            hf_repo_to_upload (str, optional): HF repo to upload the fine-tuned model to. Defaults to None.
 
         Returns:
             FinetuneResponse: Object containing information about fine-tuning job.
@@ -878,8 +883,8 @@ async def create(
             rpo_alpha=rpo_alpha,
             simpo_gamma=simpo_gamma,
             from_checkpoint=from_checkpoint,
-            from_hf_model=from_hf_model,
             hf_api_token=hf_api_token,
+            hf_repo_to_upload=hf_repo_to_upload,
         )
 
         if verbose:

src/together/types/finetune.py

@@ -212,7 +212,9 @@ class FinetuneRequest(BaseModel):
     )
     # from step
     from_checkpoint: str | None = None
+    # hf related fields
     hf_api_token: str | None = None
+    hf_repo_to_upload: str | None = None
 
 
 class FinetuneResponse(BaseModel):