Skip to content

Commit 0bce2f4

Browse files
committed
Explicit auth for finetuning
1 parent 5a01673 commit 0bce2f4

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

llm-complete-guide/steps/finetune_embeddings.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
DATASET_NAME_DISTILABEL,
2424
EMBEDDINGS_MODEL_ID_BASELINE,
2525
EMBEDDINGS_MODEL_ID_FINE_TUNED,
26-
EMBEDDINGS_MODEL_MATRYOSHKA_DIMS,
26+
EMBEDDINGS_MODEL_MATRYOSHKA_DIMS, SECRET_NAME,
2727
)
2828
from datasets import DatasetDict, concatenate_datasets, load_dataset
2929
from datasets.arrow_dataset import Dataset
@@ -288,8 +288,13 @@ def finetune(
288288
)
289289

290290
trainer.train()
291+
292+
zenml_client = Client()
293+
291294
trainer.model.push_to_hub(
292-
f"zenml/{EMBEDDINGS_MODEL_ID_FINE_TUNED}", exist_ok=True
295+
f"zenml/{EMBEDDINGS_MODEL_ID_FINE_TUNED}",
296+
exist_ok=True,
297+
token=zenml_client.get_secret(SECRET_NAME).secret_values["hf_token"]
293298
)
294299

295300
log_model_metadata(

0 commit comments

Comments
 (0)