ij.A,ddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZed dgddZeddgddZed ddZedddZ ddZy)N)logging) keras_export)legacy_h5_format) saving_lib)build_orbax_abstract_pytree)find_latest_orbax_checkpoint)is_orbax_checkpoint) file_utils)io_utils)h5py)ocpzkeras.saving.save_modelzkeras.models.save_modelc |jdd}|jdd}|r[t|jdst|jdrtjd|nt d||r%t d t |jt|jdrtjd t|jd }|| } | xrtjj|}|r|stj|} | sy |r0t|jdrtj ||S|stj ||d St|jdrt#j$||||St d|d#t$rd}YwxYw)aSaves a model as a `.keras` file. Args: model: Keras model instance to be saved. filepath: `str` or `pathlib.Path` object. Path where to save the model. overwrite: Whether we should overwrite any existing model at the target location, or instead ask the user via an interactive prompt. zipped: Whether to save the model as a zipped `.keras` archive (default when saving locally), or as an unzipped directory (default when saving on the Hugging Face Hub). Example: ```python model = keras.Sequential( [ keras.layers.Dense(5, input_shape=(3,)), keras.layers.Softmax(), ], ) model.save("model.keras") loaded_model = keras.saving.load_model("model.keras") x = keras.random.uniform((10, 3)) assert np.allclose(model.predict(x), loaded_model.predict(x)) ``` Note that `model.save()` is an alias for `keras.saving.save_model()`. The saved `.keras` file is a `zip` archive that contains: - The model's configuration (architecture) - The model's weights - The model's optimizer's state (if any) Thus models can be reinstantiated in the exact same state. include_optimizerT save_formatF.h5.hdf5.keraszThe `save_format` argument is deprecated in Keras 3. We recommend removing this argument as it can be inferred from the file path. Received: save_format=zThe `save_format` argument is deprecated in Keras 3. Please remove this argument and pass a file path with either `.keras` or `.h5` extension.Received: save_format=z-The following argument(s) are not supported: aYou are saving your model as an HDF5 file via `model.save()` or `keras.saving.save_model(model)`. This file format is considered legacy. We recommend using instead the native Keras format, e.g. `model.save('my_model.keras')` or `keras.saving.save_model(model, 'my_model.keras')`. hf://N)zippedaInvalid filepath extension for saving. Please add either a `.keras` extension for the native Keras format (recommended) or a `.h5` extension. Use `model.export(filepath)` if you want to export a SavedModel for use with TFLite/TFServing/etc. Received: filepath=.)popstrendswithrwarning ValueErrorlistkeys startswithospathexists TypeErrorr ask_to_proceed_with_overwriter save_modelrsave_model_to_hdf5) modelfilepath overwriterkwargsrris_hfr"proceeds p/var/www/html/emotional.easysim.app/public_html/venv/lib/python3.12/site-packages/keras/src/saving/saving_api.pyr%r%sL #6=**]E2K x= ! !"2 3s8}7M7M 8  OO)*5 7 )*5 7   ;FKKM"# %   8}./ C  M $ $W -E ~)9!9i88B  #h-((2$$UH55 $$UHUCC 8}./22 8Y(9     'Zq  *  s8$G GGzkeras.saving.load_modelzkeras.models.load_modelct|jdxrtj|}t j |xr)t j t j|d}t|jd}d} t j|rt j |sz|sx|svtj}t j|tjj|}t j||tj|r|}d}|s|s|r3tj |||||t#j$|dSSt|jdr3t'j(|||||t#j$|dSSt+|r)t-|||||t#j$|dSSt|jdrt/d |d t/d |d |d #|t#j$|dwwxYw)aLoads a model saved via `model.save()` or from an Orbax checkpoint. Args: filepath: `str` or `pathlib.Path` object, path to the saved model file or Orbax checkpoint directory. custom_objects: Optional dictionary mapping names (strings) to custom classes or functions to be considered during deserialization. compile: Boolean, whether to compile the model after loading. safe_mode: Boolean, whether to disallow unsafe `lambda` deserialization. When `safe_mode=False`, loading an object has the potential to trigger arbitrary code execution. This argument is only applicable to the Keras v3 model format. Defaults to `True`. Returns: A Keras model instance. If the original model was compiled, and the argument `compile=True` is set, then the returned model will be compiled. Otherwise, the model will be left uncompiled. Example: ```python model = keras.Sequential([ keras.layers.Dense(5, input_shape=(3,)), keras.layers.Softmax()]) model.save("model.keras") loaded_model = keras.saving.load_model("model.keras") ``` Note that the model variables may have different name values (`var.name` property, e.g. `"dense_1/kernel:0"`) after being reloaded. It is recommended that you use layer attributes to access specific variables, e.g. `model.get_layer("dense_1").kernel`. rz config.jsonrNTcustom_objectscompile safe_mode) ignore_errorsrzFile not found: filepath=z<. Please ensure the file is an accessible `.keras` zip file.$File format not supported: filepath=a. Keras 3 only supports V3 `.keras` files, legacy H5 format files (`.h5` extension). Note that the legacy SavedModel format is not supported by `load_model()` in Keras 3. In order to reload a TensorFlow SavedModel as an inference-only layer in Keras 3, use `keras.layers.TFSMLayer(za, call_endpoint='serving_default')` (note that your `call_endpoint` might have a different name).)rrzipfile is_zipfiler isdirr"joinris_remote_pathr get_temp_dirr r!basenamecopy load_modelshutilrmtreerload_model_from_hdf5r !_load_model_from_orbax_checkpointr) r(r0r1r2 is_keras_zip is_keras_dirr+tmp_dir local_paths r-r=r=}s:Hx=))(38J8J9L##H-*2C2C-03L M $ $W -EGA7  % %h /$$X.  --/G#"''2B2B82LMJ OOHj 1!!*-%# <5((-# X   MM' 6 M x= ! !"2 3#88-# J   MM' 6 ; x (4-# 8   MM' 6 +] # #H -+H:6  6xjA+*0 0     MM' 6 sCH(,2H(9H(-;H((Izkeras.saving.save_weightsc ht|}||jdstd|||jdstd| tjj |}|r|stj|}|sytj|||fi|y#t $rd}YBwxYw)N .weights.h5z;The filename must end in `.weights.h5`. Received: filepath=)z weights.h5z weights.jsonz`The filename must end in `.weights.json` when `max_shard_size` is specified. Received: filepath=F) rrrr r!r"r#r r$rsave_weights_only)r'r(r)max_shard_sizer* filepath_strr"r,s r- save_weightsrKsx=Ll&;&;M&J "". 1    #L,A,A&- --9N <  )i88F   .KFK sB## B10B1zkeras.saving.load_weightsc t}|jdd}|jdd}|rtd||jdr9|td|tdt j ||y|jds|jd r*|tdt j ||| y|jd s|jd r|tdt js td t jd5}d|jvrd|vrt j|d}|rtj|||ntj|||dddytrtt!j"}t%fd|D} | r t'} n} t)| |j+} t-j.| t1| } | d} |j3| ytdd#1swYyxYw)Nobjects_to_skipby_namezInvalid keyword arguments: rzF`objects_to_skip` only supports loading '.weights.h5' files.Received: zI`by_name` only supports loading legacy '.h5' or '.hdf5' files. Received: ) skip_mismatchrGz .weights.json)rOrMrrzWLoading HDF5 files requires the h5py package. You can install it via `pip install h5py`r layer_names model_weightsc3K|]?}tjtj|xr|jAyw)N)r r7r8isdigit).0itemr(s r- zload_weights..Ms=    Z__Xt< = P$,,. P sAA)pytreerXr4z. Keras 3 only supports V3 `.keras` files, `.weights.h5` files, legacy H5 format files (`.h5` extension), or Orbax checkpoints.)rrrrrload_weights_onlyr available ImportErrorFileattrssafe_get_h5_groupr$load_weights_from_hdf5_group_by_nameload_weights_from_hdf5_groupr r listdiranyrrget_state_treer load_checkpointablesdictset_state_tree)r'r(rOr*rJrMrNfitemshas_step_subdirscheckpoint_pathabstract_pytreeloaded_checkpointables loaded_states ` r- load_weightsrn sx=Ljj!2D9OjjD)G 6vh?@@X&  &%J(   $$,:/  $$ 8=    } -1F1F2  $$,:/  $$  '+    u %)>)>w)G  &%J( ~~< YYx % AGG+10D00OD EEum!==um   X &x=""8,    :8DO'O6 U113 "%!9!9 T9" .h7  \*28*=7 7  Y  s :AI  Ic:tjt|}ttj j |}tjj|}gd}tj5t|j|jj}d|vr td|j|ddi} t!j"| dd|||} t%|| j'} d| i} d |vrd| d <|j|| } | d}| j)d }ddd|Dcic] }|vr|||}} j+|t!j,| | S#1swYIxYwcc}w) aLoad a model from an Orbax checkpoint directory. `model_config` is stored as its own checkpointable (separate from `pytree`), so loading proceeds in two simple steps: 1. Load the `model_config` checkpointable to obtain the model configuration string and rebuild the model. 2. Load the `pytree` checkpointable (all arrays). When a JAX distribution is active, an abstract pytree with target shardings is provided so that Orbax reshards arrays onto the current layout. ) directory)trainable_variablesnon_trainable_variablesoptimizer_variablesmetrics_variables model_configzqCheckpoint does not contain model configuration. This checkpoint may have been saved with save_weights_only=True.Nconfigr/rXassets)r initializerintr r!r;training CheckpointerContextsetcheckpointables_metadatametadatarrrdr_model_from_configrrcgetrf_load_assets_from_dict)r(r0r1r2rjstep checkpointer variable_keys saved_keys config_loadedr'rkrequestloadedcomposite_state assets_datakey state_trees r-rArAps NN384( -- . )( 3)  6 U113 _- z ! $GH 224A *jj* I$+P!  / ! _S !!J $%%e[9 Le$+$+LsCF F F)TN)NTT)F)r r>r5abslrkeras.src.api_exportrkeras.src.legacy.savingrkeras.src.savingrkeras.src.saving.orbax_utilrrr keras.src.utilsr r keras.src.utils.module_utilsr r r%r=rKrnrAr-rs -4'CD;&$-,(*CDEfFfR(*CDEm7Fm7`)*48L+L6)*b +b L<@Rr