add fine tuned example with new params (#149)

substratusai · Jul 30, 2023 · f99d368 · f99d368
1 parent f9ff0a9
commit f99d368
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/examples/falcon-7b-instruct/finetuned-model-custom-prompt.yaml b/examples/falcon-7b-instruct/finetuned-model-custom-prompt.yaml
@@ -10,7 +10,7 @@ spec:
   trainingDataset:
     name: k8s-instructions
   params:
-    epochs: 1
+    num_train_epochs: 1
     prompt_template: |
       You're a helpful assistant that helps with generating K8s YAML manifests. Your responses should only be YAML or multi YAML files
       ## Instruction

diff --git a/examples/falcon-7b-instruct/finetuned-model.yaml b/examples/falcon-7b-instruct/finetuned-model.yaml
@@ -10,7 +10,9 @@ spec:
   trainingDataset:
     name: k8s-instructions
   params:
-    epochs: 1
+    # See HuggingFace transformers.TrainingArguments for all parameters
+    # https://huggingface.co/docs/transformers/main_classes/trainer#transformers.TrainingArguments
+    num_train_epochs: 1
   resources:
     gpu:
       count: 4