remove the hf training ckpt

Author: timofeev1995

src/together/cli/api/finetune.py

@@ -200,19 +200,11 @@ def fine_tuning(ctx: click.Context) -> None:
     "The format: {$JOB_ID/$OUTPUT_MODEL_NAME}:{$STEP}. "
     "The step value is optional, without it the final checkpoint will be used.",
 )
-@click.option(
-    "--from-hf-model",
-    type=str,
-    default=None,
-    help="Model name from the Hugging Face Hub that will be used to initialize the trained model. "
-    "The model config is not validated; any model supported by Transformers should work, but the batch size "
-    "limits are not checked.",
-)
 @click.option(
     "--hf-api-token",
     type=str,
     default=None,
-    help="HF API token to use to download a checkpoint from a private repo",
+    help="HF API token to use to upload a checkpoint to a private repo",
 )
 def create(
     ctx: click.Context,
@@ -248,7 +240,6 @@ def create(
     rpo_alpha: float | None,
     simpo_gamma: float | None,
     from_checkpoint: str,
-    from_hf_model: str,
     hf_api_token: str,
 ) -> None:
     """Start fine-tuning"""
@@ -286,7 +277,6 @@ 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,
     )
 

src/together/resources/finetune.py

@@ -76,7 +76,6 @@ def create_finetune_request(
     rpo_alpha: float | None = None,
     simpo_gamma: float | None = None,
     from_checkpoint: str | None = None,
-    from_hf_model: str | None = None,
     hf_api_token: str | None = None,
 ) -> FinetuneRequest:
     if model is not None and from_checkpoint is not None:
@@ -264,7 +263,6 @@ def create_finetune_request(
         wandb_name=wandb_name,
         training_method=training_method_cls,
         from_checkpoint=from_checkpoint,
-        from_hf_model=from_hf_model,
         hf_api_token=hf_api_token,
     )
 
@@ -345,7 +343,6 @@ def create(
         rpo_alpha: float | None = None,
         simpo_gamma: float | None = None,
         from_checkpoint: str | None = None,
-        from_hf_model: str | None = None,
         hf_api_token: str | None = None,
     ) -> FinetuneResponse:
         """
@@ -403,9 +400,6 @@ def create(
             from_checkpoint (str, optional): The checkpoint identifier to continue training from a previous fine-tuning job.
                 The format: {$JOB_ID/$OUTPUT_MODEL_NAME}:{$STEP}.
                 The step value is optional, without it the final checkpoint will be used.
-            from_hf_model (str, optional): Model name from the Hugging Face Hub that will be used to initialize the trained model.
-                The model config is not validated; any model supported by Transformers should work, but the batch size
-                limits are not checked. Defaults to None.
             hf_api_token (str, optional): API key for the Hugging Face Hub. Defaults to None.
 
         Returns:
@@ -460,7 +454,6 @@ 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,
         )
 
@@ -774,7 +767,6 @@ async def create(
         rpo_alpha: float | None = None,
         simpo_gamma: float | None = None,
         from_checkpoint: str | None = None,
-        from_hf_model: str | None = None,
         hf_api_token: str | None = None,
     ) -> FinetuneResponse:
         """
@@ -832,9 +824,6 @@ async def create(
             from_checkpoint (str, optional): The checkpoint identifier to continue training from a previous fine-tuning job.
                 The format: {$JOB_ID/$OUTPUT_MODEL_NAME}:{$STEP}.
                 The step value is optional, without it the final checkpoint will be used.
-            from_hf_model (str, optional): Model name from the Hugging Face Hub that will be used to initialize the trained model.
-                The model config is not validated; any model supported by Transformers should work, but the batch size
-                limits are not checked. Defaults to None.
             hf_api_token (str, optional): API key for the Huggging Face Hub. Defaults to None.
 
         Returns:

src/together/types/finetune.py

@@ -212,7 +212,6 @@ class FinetuneRequest(BaseModel):
     )
     # from step
     from_checkpoint: str | None = None
-    from_hf_model: str | None = None
     hf_api_token: str | None = None