From 71cb5b98c91a13d779be34c09aa7978bc0602627 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Wed, 8 Jan 2025 12:14:16 -0500
Subject: [PATCH] set lora_dropout explicitly

---
 tests/e2e/integrations/test_kd.py | 1 +
 1 file changed, 1 insertion(+)

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
         )