Fixes improper serialization code route for functional models.

PiperOrigin-RevId: 524097977

Author: nkovela1

tensorflow_graphics/projects/gan/architectures_style_gan_test.py

@@ -62,6 +62,7 @@ def test_style_based_generator_intermediate_outputs_shape(self, batch_size):
 
   def test_cloning_style_based_generator(self):
     generator, _, _ = architectures_style_gan.create_style_based_generator()
+    generator.use_legacy_config = True  # Old Keras serialization format flag
 
     with tf.keras.utils.custom_object_scope(
         architectures_style_gan.CUSTOM_LAYERS):

tensorflow_graphics/projects/gan/architectures_style_gan_v2_test.py

@@ -60,6 +60,7 @@ def test_generator_output_size_with_noise_inputs(self):
 
   def test_cloning_generator(self):
     generator, _, _ = architectures_style_gan_v2.create_style_based_generator()
+    generator.use_legacy_config = True  # Old Keras serialization format flag
 
     with tf.keras.utils.custom_object_scope(
         architectures_style_gan_v2.CUSTOM_LAYERS):