diff --git a/tests/e2e/integrations/test_kd.py b/tests/e2e/integrations/test_kd.py index 919a73dce..c23586680 100644 --- 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 )