We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5a01673 commit 0bce2f4Copy full SHA for 0bce2f4
llm-complete-guide/steps/finetune_embeddings.py
@@ -23,7 +23,7 @@
23
DATASET_NAME_DISTILABEL,
24
EMBEDDINGS_MODEL_ID_BASELINE,
25
EMBEDDINGS_MODEL_ID_FINE_TUNED,
26
- EMBEDDINGS_MODEL_MATRYOSHKA_DIMS,
+ EMBEDDINGS_MODEL_MATRYOSHKA_DIMS, SECRET_NAME,
27
)
28
from datasets import DatasetDict, concatenate_datasets, load_dataset
29
from datasets.arrow_dataset import Dataset
@@ -288,8 +288,13 @@ def finetune(
288
289
290
trainer.train()
291
+
292
+ zenml_client = Client()
293
294
trainer.model.push_to_hub(
- 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"]
298
299
300
log_model_metadata(
0 commit comments