From 1366dcf479da6fe3d0d80c199ef8db2fd33466a4 Mon Sep 17 00:00:00 2001 From: "yuze.zyz" Date: Tue, 10 Dec 2024 19:28:51 +0800 Subject: [PATCH] fix ut --- tests/trainers/test_finetune_vision_efficient_tuning_swift.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/trainers/test_finetune_vision_efficient_tuning_swift.py b/tests/trainers/test_finetune_vision_efficient_tuning_swift.py index 56a5b6fc..6ab6211c 100644 --- a/tests/trainers/test_finetune_vision_efficient_tuning_swift.py +++ b/tests/trainers/test_finetune_vision_efficient_tuning_swift.py @@ -51,13 +51,14 @@ class TestVisionEfficientTuningSwiftTrainer(unittest.TestCase): cfg.model.finetune = True cfg.train.max_epochs = self.max_epochs cfg.train.lr_scheduler.T_max = self.max_epochs + cfg.train.dataloader.workers_per_gpu = 0 + cfg.evaluation.dataloader.workers_per_gpu = 0 cfg.model.backbone.lora_length = 0 return cfg lora_config = LoRAConfig( r=self.tune_length, target_modules=['qkv'], - merge_weights=False, use_merged_linear=True, enable_lora=[True])