[Example] Fix llm_finetune example json part

ludwig-ai · Jul 13, 2023 · bcdf349 · bcdf349
1 parent 60f1416
commit bcdf349
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/examples/llm_finetuning/train_imdb_ray.py b/examples/llm_finetuning/train_imdb_ray.py
@@ -11,17 +11,18 @@
   - name: review
     type: text
 
-    encoder:
-      type: auto_transformer
-      pretrained_model_name_or_path: bigscience/bloom-3b
-      trainable: true
-      adapter: lora
-
 output_features:
   - name: sentiment
     type: category
 
+model_type: llm
+base_model: bigscience/bloom-3b
+
+adapter:
+  type: lora
+
 trainer:
+  type: finetune
   batch_size: 4
   epochs: 3