fix ut (#1143)

2026-09-01 19:49:03 +02:00 · 2024-12-11 10:19:29 +08:00
parent 40b689795a
commit 7579fac6f3
1 changed files with 2 additions and 1 deletions
--- 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])