@@ -50,6 +50,7 @@ def test_basic(self):
5050 sft_args = SftArguments (
5151 model_type = model_type ,
5252 template_type = 'AUTO' ,
53+ lora_target_modules = 'ALL' ,
5354 quantization_bit = quantization_bit ,
5455 batch_size = 2 ,
5556 eval_steps = 5 ,
@@ -138,6 +139,7 @@ def test_vl_audio(self):
138139 template_type = 'AUTO' ,
139140 eval_steps = 5 ,
140141 check_dataset_strategy = 'warning' ,
142+ lora_target_modules = 'ALL' ,
141143 train_dataset_sample = 200 ,
142144 dataset = [dataset ],
143145 output_dir = output_dir ,
@@ -245,6 +247,7 @@ def test_cogagent_instruct(self):
245247 model_type = ModelType .cogagent_18b_instruct ,
246248 dataset = DatasetName .coco_mini_en_2 ,
247249 train_dataset_sample = 100 ,
250+ lora_target_modules = 'ALL' ,
248251 eval_steps = 5 ,
249252 quantization_bit = 4 ))
250253 best_model_checkpoint = output ['best_model_checkpoint' ]
@@ -263,6 +266,7 @@ def test_xcomposer_chat(self):
263266 SftArguments (
264267 model_type = ModelType .internlm_xcomposer2_7b_chat ,
265268 dataset = DatasetName .coco_mini_en ,
269+ lora_target_modules = 'DEFAULT' ,
266270 train_dataset_sample = 100 ,
267271 eval_steps = 5 ))
268272 best_model_checkpoint = output ['best_model_checkpoint' ]
@@ -282,6 +286,7 @@ def test_yi_vl_6b_chat(self):
282286 SftArguments (
283287 model_type = ModelType .yi_vl_6b_chat ,
284288 # dataset=DatasetName.capcha_images,
289+ lora_target_modules = 'ALL' ,
285290 train_dataset_sample = 100 ,
286291 eval_steps = 5 ,
287292 custom_train_dataset_path = [
@@ -303,6 +308,7 @@ def test_dpo(self):
303308 output = dpo_main (
304309 DPOArguments (
305310 model_type = ModelType .qwen_1_8b_chat ,
311+ sft_type = 'full' ,
306312 dataset = DatasetName .hh_rlhf ,
307313 train_dataset_sample = 100 ,
308314 eval_steps = 5 ))
0 commit comments