class_docstr: Base class for all models.
  
  [`FlaxPreTrainedModel`] takes care of storing the configuration of the models and handles methods for loading,
  downloading and saving models.
  
  Class attributes (overridden by derived classes):
  
      - **config_class** ([`PretrainedConfig`]) -- A subclass of [`PretrainedConfig`] to use as configuration class
        for this model architecture.
      - **base_model_prefix** (`str`) -- A string indicating the attribute associated to the base model in derived
        classes of the same architecture adding modules on top of the base model.
      - **main_input_name** (`str`) -- The name of the principal input to the model (often `input_ids` for NLP
        models, `pixel_values` for vision models and `input_values` for speech models).
class_docstr_tok: ["Base","class","for","all","models",".","[","`","FlaxPreTrainedModel","`","]","takes","care","of","storing","the","configuration","of","the","models","and","handles","methods","for","loading",",","downloading","and","saving","models",".","Class","attributes","(","overridden","by","derived","classes",")",":","-","*","*","config_class","*","*","(","[","`","PretrainedConfig","`","]",")","--","A","subclass","of","[","`","PretrainedConfig","`","]","to","use","as","configuration","class","for","this","model","architecture",".","-","*","*","base_model_prefix","*","*","(","`","str","`",")","--","A","string","indicating","the","attribute","associated","to","the","base","model","in","derived","classes","of","the","same","architecture","adding","modules","on","top","of","the","base","model",".","-","*","*","main_input_name","*","*","(","`","str","`",")","--","The","name","of","the","principal","input","to","the","model","(","often","`","input_ids","`","for","NLP","models",",","`","pixel_values","`","for","vision","models","and","`","input_values","`","for","speech","models",")","."]
code: def params(self, params: Union[Dict, FrozenDict]):
      # don't set params if the model is not initialized
      if not self._is_initialized:
          raise ValueError(
              "`params` cannot be set from model when the model is created with `_do_init=False`. "
              "You store the params outside of the model."
          )
  
      if isinstance(params, FrozenDict):
          params = unfreeze(params)
      param_keys = set(flatten_dict(params).keys())
      if len(self.required_params - param_keys) > 0:
          raise ValueError(
              "Some parameters are missing. Make sure that `params` include the following "
              f"parameters {self.required_params - param_keys}"
          )
      self._params = params
  
imports: import gc
  import json
  import os
  import re
  from functools import partial
  from pickle import UnpicklingError
  from typing import Any, Dict, Set, Tuple, Union
  import flax.linen
  import jax
  import jax.numpy
  import msgpack.exceptions
  from flax.core.frozen_dict import FrozenDict, unfreeze
  from flax.serialization import from_bytes, to_bytes
  from flax.traverse_util import flatten_dict, unflatten_dict
  from jax.random import PRNGKey
  from .configuration_utils import PretrainedConfig
  from .dynamic_module_utils import custom_object_save
  from .generation_flax_utils import FlaxGenerationMixin
  from .modeling_flax_pytorch_utils import load_pytorch_checkpoint_in_flax_state_dict
  from .utils import FLAX_WEIGHTS_INDEX_NAME, FLAX_WEIGHTS_NAME, WEIGHTS_INDEX_NAME, WEIGHTS_NAME, PushToHubMixin, add_code_sample_docstrings, add_start_docstrings_to_model_forward, cached_file, copy_func, download_url, has_file, is_offline_mode, is_remote_url, logging, replace_return_docstrings
  from .utils.hub import convert_file_size_to_int, get_checkpoint_shard_files
label_desc: Use image node_id 11 for calling the FlaxPreTrainedModel obj's underlying member method code with example usage: obj.params(params) without return types