We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1610ab1 commit 41d618bCopy full SHA for 41d618b
src/configs.js
@@ -298,7 +298,6 @@ export function getCacheShapes(config, options) {
298
/** @type {Record<string, number[]>} */
299
const cache_values = {};
300
301
- // @ts-expect-error TS2339
302
const {
303
layer_types,
304
num_attention_heads,
@@ -310,7 +309,7 @@ export function getCacheShapes(config, options) {
310
309
mamba_d_state,
311
mamba_n_groups,
312
mamba_expand,
313
- } = config;
+ } = /** @type {any} */(config);
314
const head_dim = hidden_size / num_attention_heads;
315
const batch_size = options?.batch_size ?? 1;
316
const conv_d_inner = mamba_expand * hidden_size + 2 * mamba_n_groups * mamba_d_state;
0 commit comments