From 85f4b4207f3f2ba73c86764576c9e4a6bfa7732b Mon Sep 17 00:00:00 2001
From: Chongxiao Cao <chongxiaoc@uber.com>
Date: Wed, 12 Jul 2023 17:24:09 -0700
Subject: [PATCH] [Example] Fix llm_finetune example json part

---
 examples/llm_finetuning/train_imdb_ray.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/llm_finetuning/train_imdb_ray.py b/examples/llm_finetuning/train_imdb_ray.py
index 43dfc3ae0ef..f28b28f4d9c 100644
--- a/examples/llm_finetuning/train_imdb_ray.py
+++ b/examples/llm_finetuning/train_imdb_ray.py
@@ -22,6 +22,7 @@
     type: category
 
 trainer:
+  type: finetune
   batch_size: 4
   epochs: 3