Dec 5, 2025 · Dec 8, 2025 · Dec 8, 2025 · Dec 8, 2025 · Dec 8, 2025 · Dec 8, 2025
diff --git a/keras/src/saving/saving_api.py b/keras/src/saving/saving_api.py

 def _is_orbax_checkpoint(filepath):
    """Check if the given path is an Orbax checkpoint directory."""
    if not os.path.exists(filepath):
        return False

    try:
        from keras.src.utils.module_utils import ocp
Original file line number	Diff line number	Diff line change
Expand Up		@@ -130,6 +130,9 @@ def _load_model_from_orbax_checkpoint(

		def _is_orbax_checkpoint(filepath):
amitsrivastava78 marked this conversation as resolved. Show resolvedHide resolved
		"""Check if the given path is an Orbax checkpoint directory."""
		if not os.path.exists(filepath):
		return False

		try:
		from keras.src.utils.module_utils import ocp

Expand Down