You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: content/en/docs/components/training/user-guides/fine-tuning.md
+13Lines changed: 13 additions & 0 deletions
Original file line number
Diff line number
Diff line change
@@ -87,6 +87,19 @@ TrainingClient().train(
87
87
After you execute `train`, the Training Operator will orchestrate the appropriate PyTorchJob resources
88
88
to fine-tune the LLM.
89
89
90
+
## Using custom images with Fine-Tuning API
91
+
Platform engineers can customize the storage initializer and trainer images by setting the `STORAGE_INITIALIZER_IMAGE` and `TRAINER_TRANSFORMER_IMAGE` environment variables before executing the `train` command.
92
+
93
+
For example: In your python code, set the env vars before executing `train`:
- Run the example to [fine-tune the TinyLlama LLM](https://github.com/kubeflow/training-operator/blob/6ce4d57d699a76c3d043917bd0902c931f14080f/examples/pytorch/language-modeling/train_api_hf_dataset.ipynb)
0 commit comments