From 7579fac6f3f3f2601adc6e14bce5b103d4d2e256 Mon Sep 17 00:00:00 2001
From: tastelikefeet <58414341+tastelikefeet@users.noreply.github.com>
Date: Wed, 11 Dec 2024 10:19:29 +0800
Subject: [PATCH] fix ut (#1143)

---
 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])