From 881143cbe895a9fe47a293e4cb2175d3c5342e49 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sat, 12 Sep 2026 21:00:31 -0600 Subject: [PATCH] Fixed issue with EMA climbing decay instead of reducing it. --- toolkit/ema.py | 2 +- version.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/toolkit/ema.py b/toolkit/ema.py index 939e1f8..dac8ef6 100644 --- a/toolkit/ema.py +++ b/toolkit/ema.py @@ -133,7 +133,7 @@ class ExponentialMovingAverage: if param.dtype != torch.float32: param_float = param_float.to(torch.float32) gap = (s_param_float - param_float) - s_param_float.add_(gap * one_minus_decay) + s_param_float.sub_(gap * one_minus_decay) update_param = False if self.use_feedback: diff --git a/version.py b/version.py index 41472ec..40d5e46 100644 --- a/version.py +++ b/version.py @@ -1 +1 @@ -VERSION = "0.13.8" +VERSION = "0.13.9"