[Example] Fix llm_finetune example json part

ludwig-ai · Jul 13, 2023 · 6ffb182 · 6ffb182
1 parent 60f1416
commit 6ffb182
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/examples/llm_finetuning/train_imdb_ray.py b/examples/llm_finetuning/train_imdb_ray.py
@@ -15,12 +15,15 @@
       type: auto_transformer
       pretrained_model_name_or_path: bigscience/bloom-3b
       trainable: true
-      adapter: lora
 
 output_features:
   - name: sentiment
     type: category
 
+
+adapter:
+  type: lora
+
 trainer:
   batch_size: 4
   epochs: 3