Skip to content

Commit 41d618b

Browse files
committed
Fix type error
1 parent 1610ab1 commit 41d618b

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/configs.js

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -298,7 +298,6 @@ export function getCacheShapes(config, options) {
298298
/** @type {Record<string, number[]>} */
299299
const cache_values = {};
300300

301-
// @ts-expect-error TS2339
302301
const {
303302
layer_types,
304303
num_attention_heads,
@@ -310,7 +309,7 @@ export function getCacheShapes(config, options) {
310309
mamba_d_state,
311310
mamba_n_groups,
312311
mamba_expand,
313-
} = config;
312+
} = /** @type {any} */(config);
314313
const head_dim = hidden_size / num_attention_heads;
315314
const batch_size = options?.batch_size ?? 1;
316315
const conv_d_inner = mamba_expand * hidden_size + 2 * mamba_n_groups * mamba_d_state;

0 commit comments

Comments
 (0)