Update pytorch-metadata.json

lutzroeder · Aug 1, 2024 · d9416c2 · d9416c2
1 parent fca2e70
commit d9416c2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/source/pytorch-metadata.json b/source/pytorch-metadata.json
@@ -11818,7 +11818,8 @@
       { "name": "attn_mask", "type": "Tensor", "optional": true, "default": null },
       { "name": "dropout_p", "type": "float32", "default": 0.0 },
       { "name": "is_causal", "type": "boolean", "default": false },
-      { "name": "scale", "type": "float32", "optional": true, "default": null }
+      { "name": "scale", "type": "float32", "optional": true, "default": null },
+      { "name": "enable_gqa", "type": "boolean", "default": false }
     ],
     "outputs": [
       { "type": "Tensor" }