Unable to save model

Start with the linked Colab notebook and reproduce the model construction and model.save call under TensorFlow 2.11.0. Inspect the save failure around the GDN layers and Keras Sequential model; done means establishing the reproducible behavior and validating that the model can be saved successfully.

I'm creating a model to decompress my image in TF using Conv2DTranspose and GDN layer. My architecture looks like this:

import os
import tensorflow as tf
from tensorflow_compression import layers
from tensorflow.keras.layers import Conv2DTranspose
from tensorflow.keras.models import Sequential
network_channels = 128
compression_channels = 192
in_channels = 3

gdn_layer_1 = layers.GDN(inverse=True, rectify=False, data_format = "channels_first") 
gdn_layer_2 = layers.GDN(inverse=True, rectify=False, data_format = "channels_first")
gdn_layer_3 = layers.GDN(inverse=True, rectify=False, data_format = "channels_first")
model = Sequential()
model.add(Conv2DTranspose(network_channels, (5, 5), strides = (2, 2), padding= 'same', output_padding = (1,1), data_format = 'channels_first', input_shape=( compression_channels, 1, 1)))
model.add(gdn_layer_1)
model.add(Conv2DTranspose(network_channels, (5, 5), strides = (2, 2), padding= 'same', output_padding = (1,1), data_format = 'channels_first'))
model.add(gdn_layer_2)
model.add(Conv2DTranspose(network_channels, (5, 5), strides = (2, 2), padding= 'same', output_padding = (1,1), data_format = 'channels_first'))
model.add(gdn_layer_3)
model.add(Conv2DTranspose(in_channels, (5, 5), strides = (2, 2), padding= 'same', output_padding = (1,1), data_format = 'channels_first'))
model.summary()

When I'm saving my model using the below command.

model.save(os.path.join("./tf_model"))

It's resulting in this issue:

My TF version is 2.11.0. I'm also attaching link to Colab Notebook to replicate the issue.