瀏覽代碼

fix export_onnx.py

lyuxiang.lx 1 年之前
父節點
當前提交
efe1d15960
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      cosyvoice/bin/export_onnx.py

+ 1 - 1
cosyvoice/bin/export_onnx.py

@@ -171,7 +171,7 @@ def main():
                                                       sess_options=option, providers=providers)
 
         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.pop('offset')
             cache = {k: v[0] for k, v in cache.items()}