diff --git a/MindIE/MindIE-Torch/built-in/audio/mindie_whisperx/readme.md b/MindIE/MindIE-Torch/built-in/audio/mindie_whisperx/readme.md index 28b4f556e73dd5417d5cff7b149c101faac26a4a..d2a3ac3868c3da913b46de9c378a52ec91dd71a8 100644 --- a/MindIE/MindIE-Torch/built-in/audio/mindie_whisperx/readme.md +++ b/MindIE/MindIE-Torch/built-in/audio/mindie_whisperx/readme.md @@ -88,10 +88,10 @@ - -save_path: 编译好的模型的保存文件,必选。 - -device_id: 选在模型运行的卡编号,默认值0,可选。 - -soc_version: 芯片类型,必选。 - - -hardware: 机器型号,默认值800IA2,可选["300IPro", "800IA2"]。 + - -hardware: 机器型号,默认值800IA2,可选["300IDuo", "800IA2"]。 约束说明: 1. 当前暂不支持动态batch,batch_size改变后,需要重新编图。 - 2. 支持的hardware类型为"300IPro"或"800IA2"。 + 2. 支持的hardware类型为"300IDuo"或"800IA2"。 3. 芯片类型需要用户在环境上查询得到。 如果无法确定当前设备的soc_version,则在安装NPU驱动包的服务器执行npu-smi info命令进行查询, 在查询到的“Name”前增加Ascend信息,例如“Name”对应取值为xxxyy,实际配置的soc_version值为Ascendxxxyy。 diff --git a/MindIE/MindIE-Torch/built-in/audio/mindie_whisperx/whisper_large_v3/utils.py b/MindIE/MindIE-Torch/built-in/audio/mindie_whisperx/whisper_large_v3/utils.py index f775edcee6832ec1d08a1ef9e8e0dd508af3883c..0ac7106c40e72d08f5593457955354bf28e82d0d 100644 --- a/MindIE/MindIE-Torch/built-in/audio/mindie_whisperx/whisper_large_v3/utils.py +++ b/MindIE/MindIE-Torch/built-in/audio/mindie_whisperx/whisper_large_v3/utils.py @@ -31,8 +31,8 @@ class CompileInfo: encoder_seq_len = 1500 hidden_size = 1280 layer_nums = 32 - machine_type = ["300IPro", "800IA2"] - attention_type = {"300IPro": "FA_HIGH_PERF", "800IA2": "PFA"} + machine_type = ["300IDuo", "800IA2"] + attention_type = {"300IDuo": "FA_HIGH_PERF", "800IA2": "PFA"} # 2 : decoder_input_ids, encoder_outputs param_num_min = 2 # 131 : ecoder_input_ids, encoder_outputs, actual_seq_len and 128 kvcache tensors.