From 85bad57df37aef8abe0f0f4f3176a1ad56542f89 Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Wed, 13 Aug 2025 11:39:40 -0600
Subject: [PATCH] Fix bug that would use EMA when set false

---
 toolkit/config_modules.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/toolkit/config_modules.py b/toolkit/config_modules.py
index 8d9e734b..a825830b 100644
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -462,7 +462,7 @@ class TrainConfig:
 
         ema_config: Union[Dict, None] = kwargs.get('ema_config', None)
         # if it is set explicitly to false, leave it false. 
-        if ema_config is not None and ema_config.get('use_ema', None) is not None:
+        if ema_config is not None and ema_config.get('use_ema', False):
             ema_config['use_ema'] = True
             print(f"Using EMA")
         else: