Use std::optional for mask_arr arg (#2763)

2025-11-17 10:43:33 +09:00
parent 472c43a0c8
commit 32b18d8b66
3 changed files with 15 additions and 26 deletions
@@ -213,11 +213,11 @@ void init_fast(nb::module_& parent_module) {
              throw std::invalid_argument(msg.str());
            }
            return mx::fast::scaled_dot_product_attention(
-                queries, keys, values, scale, mask_str, {}, sinks, s);
+                queries, keys, values, scale, mask_str, std::nullopt, sinks, s);
          } else {
            auto mask_arr = std::get<mx::array>(mask);
            return mx::fast::scaled_dot_product_attention(
-                queries, keys, values, scale, "", {mask_arr}, sinks, s);
+                queries, keys, values, scale, "", mask_arr, sinks, s);
          }

        } else {