From d5c4aa0c03ce8187b9680e963103f179501709f5 Mon Sep 17 00:00:00 2001 From: mrq Date: Fri, 28 Feb 2025 22:13:48 -0600 Subject: [PATCH] lol --- vall_e/engines/deepspeed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vall_e/engines/deepspeed.py b/vall_e/engines/deepspeed.py index f07705b..85ec63b 100755 --- a/vall_e/engines/deepspeed.py +++ b/vall_e/engines/deepspeed.py @@ -47,7 +47,7 @@ class Engine(DeepSpeedEngine): } # kwargs['stats'] = None will return None when popped - maybe_stats = kwargs.get('stats', stats) + maybe_stats = kwargs.pop('stats', stats) if maybe_stats is not None: stats = maybe_stats