set lora_dropout explicitly

2025-01-08 12:14:16 -05:00
parent 5a1c1b82d4
commit 800e7fa41e
1 changed files with 1 additions and 0 deletions
--- a/tests/e2e/integrations/test_kd.py
+++ b/tests/e2e/integrations/test_kd.py
@@ -101,6 +101,7 @@ class TestKnowledgeDistillation:
                "lora_target_linear": True,
                "lora_r": 16,
                "lora_alpha": 32,
                "lora_dropout": 0.0,
            }
            | kd_min_cfg
        )