deepgeek/CosyVoice
2
0
Fork 0
mirror of https://github.com/FunAudioLLM/CosyVoice.git synced 2026-02-05 18:09:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
lyuxiang.lx
2025-02-11 15:56:20 +08:00
parent f77c6a85aa
commit 890300513c
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
cosyvoice/bin/average_model.py
View File

@@ -75,10 +75,11 @@ def main():
print('Processing {}'.format(path)) print('Processing {}'.format(path))
states = torch.load(path, map_location=torch.device('cpu')) states = torch.load(path, map_location=torch.device('cpu'))
for k in states.keys(): for k in states.keys():
if k not in avg.keys() and k not in ['step', 'epoch']: if k not in ['step', 'epoch']:
avg[k] = states[k].clone() if k not in avg.keys():
else: avg[k] = states[k].clone()
avg[k] += states[k] else:
avg[k] += states[k]
# average # average
for k in avg.keys(): for k in avg.keys():
if avg[k] is not None: if avg[k] is not None: