Skip to content

Commit 9e6bbc1

Browse files
committed
Fix naming
1 parent 65c7ae9 commit 9e6bbc1

2 files changed

Lines changed: 9 additions & 9 deletions

File tree

src/together/resources/finetune.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def create_finetune_request(
7777
simpo_gamma: float | None = None,
7878
from_checkpoint: str | None = None,
7979
hf_api_token: str | None = None,
80-
hf_repo_to_upload: str | None = None,
80+
hf_output_repo_name: str | None = None,
8181
) -> FinetuneRequest:
8282
if model is not None and from_checkpoint is not None:
8383
raise ValueError(
@@ -265,7 +265,7 @@ def create_finetune_request(
265265
training_method=training_method_cls,
266266
from_checkpoint=from_checkpoint,
267267
hf_api_token=hf_api_token,
268-
hf_repo_to_upload=hf_repo_to_upload,
268+
hf_output_repo_name=hf_output_repo_name,
269269
)
270270

271271
return finetune_request
@@ -346,7 +346,7 @@ def create(
346346
simpo_gamma: float | None = None,
347347
from_checkpoint: str | None = None,
348348
hf_api_token: str | None = None,
349-
hf_repo_to_upload: str | None = None,
349+
hf_output_repo_name: str | None = None,
350350
) -> FinetuneResponse:
351351
"""
352352
Method to initiate a fine-tuning job
@@ -404,7 +404,7 @@ def create(
404404
The format: {$JOB_ID/$OUTPUT_MODEL_NAME}:{$STEP}.
405405
The step value is optional, without it the final checkpoint will be used.
406406
hf_api_token (str, optional): API key for the Hugging Face Hub. Defaults to None.
407-
hf_repo_to_upload (str, optional): HF repo to upload the fine-tuned model to. Defaults to None.
407+
hf_output_repo_name (str, optional): HF repo to upload the fine-tuned model to. Defaults to None.
408408
409409
Returns:
410410
FinetuneResponse: Object containing information about fine-tuning job.
@@ -459,7 +459,7 @@ def create(
459459
simpo_gamma=simpo_gamma,
460460
from_checkpoint=from_checkpoint,
461461
hf_api_token=hf_api_token,
462-
hf_repo_to_upload=hf_repo_to_upload,
462+
hf_output_repo_name=hf_output_repo_name,
463463
)
464464

465465
if verbose:
@@ -773,7 +773,7 @@ async def create(
773773
simpo_gamma: float | None = None,
774774
from_checkpoint: str | None = None,
775775
hf_api_token: str | None = None,
776-
hf_repo_to_upload: str | None = None,
776+
hf_output_repo_name: str | None = None,
777777
) -> FinetuneResponse:
778778
"""
779779
Async method to initiate a fine-tuning job
@@ -831,7 +831,7 @@ async def create(
831831
The format: {$JOB_ID/$OUTPUT_MODEL_NAME}:{$STEP}.
832832
The step value is optional, without it the final checkpoint will be used.
833833
hf_api_token (str, optional): API key for the Huggging Face Hub. Defaults to None.
834-
hf_repo_to_upload (str, optional): HF repo to upload the fine-tuned model to. Defaults to None.
834+
hf_output_repo_name (str, optional): HF repo to upload the fine-tuned model to. Defaults to None.
835835
836836
Returns:
837837
FinetuneResponse: Object containing information about fine-tuning job.
@@ -886,7 +886,7 @@ async def create(
886886
simpo_gamma=simpo_gamma,
887887
from_checkpoint=from_checkpoint,
888888
hf_api_token=hf_api_token,
889-
hf_repo_to_upload=hf_repo_to_upload,
889+
hf_output_repo_name=hf_output_repo_name,
890890
)
891891

892892
if verbose:

src/together/types/finetune.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -214,7 +214,7 @@ class FinetuneRequest(BaseModel):
214214
from_checkpoint: str | None = None
215215
# hf related fields
216216
hf_api_token: str | None = None
217-
hf_repo_to_upload: str | None = None
217+
hf_output_repo_name: str | None = None
218218

219219

220220
class FinetuneResponse(BaseModel):

0 commit comments

Comments
 (0)