diff --git a/mediapipe/tasks/web/genai/llm_inference/llm_inference.ts b/mediapipe/tasks/web/genai/llm_inference/llm_inference.ts
index b2a11c1861..a28402c2e3 100644
--- a/mediapipe/tasks/web/genai/llm_inference/llm_inference.ts
+++ b/mediapipe/tasks/web/genai/llm_inference/llm_inference.ts
@@ -344,6 +344,8 @@ export class LlmInference extends TaskRunner {
       requiredLimits: {
         'maxStorageBufferBindingSize': MAX_STORAGE_BUFFER_BINDING_SIZE_FOR_LLM,
         'maxBufferSize': maxBufferSize,
+        'maxStorageBuffersPerShaderStage':
+          adapter.limits.maxStorageBuffersPerShaderStage,
       },
     };