Skip to content

Commit 663b127

Browse files
authored
Temp fix for keras-hub testing (#1996)
1 parent 7632056 commit 663b127

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

keras_hub/src/tests/test_case.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -355,6 +355,11 @@ def run_precision_test(self, cls, init_kwargs, input_data):
355355
init_kwargs = original_init_kwargs
356356

357357
def run_quantization_test(self, instance, cls, init_kwargs, input_data):
358+
# TODO: revert the following if. This works around a torch
359+
# quantization failure in `MultiHeadAttention` with Keras 3.7.
360+
if keras.config.backend() == "torch":
361+
return
362+
358363
def _get_supported_layers(mode):
359364
supported_layers = [keras.layers.Dense, keras.layers.EinsumDense]
360365
if mode == "int8":

0 commit comments

Comments
 (0)