diff --git a/docs_new/src/snippets/autoregressive/qwen36-deployment.jsx b/docs_new/src/snippets/autoregressive/qwen36-deployment.jsx index fd8e093de..d825dcca3 100644 --- a/docs_new/src/snippets/autoregressive/qwen36-deployment.jsx +++ b/docs_new/src/snippets/autoregressive/qwen36-deployment.jsx @@ -222,12 +222,9 @@ export const Qwen36Deployment = () => { } } - if (hardware === 'b200') { + if (hardware === 'b200' || hardware === 'b300') { cmd += ` \\\n --attention-backend trtllm_mha`; } - if (hardware === 'b300') { - cmd += ` \\\n --attention-backend flashinfer`; - } if (hwConfig.mem !== undefined) { cmd += ` \\\n --mem-fraction-static ${hwConfig.mem}`; }