No public description

PiperOrigin-RevId: 681659640
google-ai-edge · Oct 3, 2024 · 8dc0230 · 8dc0230
1 parent 4681a0a
commit 8dc0230
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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,
       },
     };