mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

lol

Browse Source
This commit is contained in:
mrq 2025-02-28 22:13:48 -06:00
parent ddc49c89c5
commit d5c4aa0c03
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vall_e/engines/deepspeed.py
View File

@ -47,7 +47,7 @@ class Engine(DeepSpeedEngine):
} }
# kwargs['stats'] = None will return None when popped # 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: if maybe_stats is not None:
stats = maybe_stats stats = maybe_stats