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

fix export_onnx.py

Browse Source
This commit is contained in:
lyuxiang.lx
2025-04-15 16:51:25 +08:00
parent 9ebcf7b1ad
commit efe1d15960
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cosyvoice/bin/export_onnx.py
View File

@@ -171,7 +171,7 @@ def main():
sess_options=option, providers=providers) sess_options=option, providers=providers)
for _ in tqdm(range(10)): for _ in tqdm(range(10)):
x, mask, mu, t, spks, cond = get_dummy_input(batch_size, random.randint(16, 512), out_channels, device) x, mask, mu, t, spks, cond = get_dummy_input(batch_size, random.randint(16, 256), out_channels, device)
cache = model.model.init_flow_cache()['decoder_cache'] cache = model.model.init_flow_cache()['decoder_cache']
cache.pop('offset') cache.pop('offset')
cache = {k: v[0] for k, v in cache.items()} cache = {k: v[0] for k, v in cache.items()}