From f23ccab77cbce030df17a4ce8a4f5426fec19bb6 Mon Sep 17 00:00:00 2001 From: harshaljanjani Date: Wed, 24 Sep 2025 15:56:53 +0530 Subject: [PATCH] fix: Add compute_output_shape() for Moonshine JAX nightly --- keras_hub/src/models/moonshine/moonshine_backbone.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/keras_hub/src/models/moonshine/moonshine_backbone.py b/keras_hub/src/models/moonshine/moonshine_backbone.py index ff6b4f2756..28ff3d1a3d 100644 --- a/keras_hub/src/models/moonshine/moonshine_backbone.py +++ b/keras_hub/src/models/moonshine/moonshine_backbone.py @@ -54,6 +54,9 @@ def call(self, inputs): sequence_length = keras.ops.shape(inputs)[1] return keras.ops.arange(sequence_length, dtype="int32") + def compute_output_shape(self, input_shape): + return (input_shape[1],) + @keras_hub_export("keras_hub.models.MoonshineBackbone") class MoonshineBackbone(Backbone):