Skip to content

Commit 1d0ea32

Browse files
authored
Add YAML example for fine-tuning job
1 parent 1f75c53 commit 1d0ea32

File tree

1 file changed

+28
-3
lines changed

1 file changed

+28
-3
lines changed

articles/ai-studio/how-to/fine-tune-serverless.md

Lines changed: 28 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -193,9 +193,34 @@ You can use this [sample notebook](https://github.com/Azure/azureml-examples/blo
193193

194194
Additionally, you can use this sample CLI to create a standalone fine-tuning job to enhance a model's ability to summarize dialogues between two people using a dataset.
195195

196-
::: zone pivot="programming-language-cli"
197-
[!INCLUDE [cli](cli/jobs/finetuning/standalone/model-as-a-service/chat-completion/chat-completion-finetuning.yaml)]
198-
::: zone-end
196+
```yaml
197+
type: finetuning
198+
199+
name: "Phi-3-mini-4k-instruct-with-amlcompute"
200+
experiment_name: "Phi-3-mini-4k-instruct-finetuning-experiment"
201+
display_name: "Phi-3-mini-4k-instruct-display-name"
202+
task: chat_completion
203+
model_provider: custom
204+
model:
205+
path: "azureml://registries/azureml/models/Phi-3-mini-4k-instruct/versions/14"
206+
type: mlflow_model
207+
training_data: train.jsonl
208+
validation_data:
209+
path: validation.jsonl
210+
type: uri_file
211+
hyperparameters:
212+
num_train_epochs: "1"
213+
per_device_train_batch_size: "1"
214+
learning_rate: "0.00002"
215+
properties:
216+
my_property: "my_value"
217+
tags:
218+
foo_tag: "bar"
219+
outputs:
220+
registered_model:
221+
name: "Phi-3-mini-4k-instruct-finetuned-model"
222+
type: mlflow_model
223+
```
199224
200225
The training data used is the same as demonstrated in the SDK notebook. The CLI employs the available Azure AI models for the chat-completion task. If you prefer to use a different model than the one in the CLI sample, you can update the arguments, such as 'model path,' accordingly
201226

0 commit comments

Comments
 (0)