change new in functional

Author: divyashreepathihalli

keras/src/models/functional.py

@@ -98,11 +98,8 @@ class Functional(Function, Model):
     """
 
     def __new__(cls, *args, **kwargs):
-        if is_nnx_enabled():
-            from flax import nnx
-            instance = super(Functional, cls).__new__(cls)
-            vars(instance)["_object__state"] = nnx.object.ObjectState()
-        return typing.cast(cls, super().__new__(cls))
+        obj = super().__new__(cls, *args, **kwargs)
+        return obj
 
 
     @tracking.no_automatic_dependency_tracking