| funasr/utils/export_utils.py | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 |
funasr/utils/export_utils.py
@@ -78,7 +78,7 @@ ) def _torchscripts(model, path, device='cpu'): def _torchscripts(model, path, device='cuda'): dummy_input = model.export_dummy_inputs() if device == 'cuda':
