class_docstr: Mixin containing common functions for the schedulers.
  
  Class attributes:
      - **_compatibles** (`List[str]`) -- A list of classes that are compatible with the parent class, so that
        `from_config` can be used from a class different than the one used to save the config (should be overridden
        by parent class).
class_docstr_tok: ["Mixin","containing","common","functions","for","the","schedulers",".","Class","attributes",":","-","*","*","_compatibles","*","*","(","`","List","[","str","]","`",")","--","A","list","of","classes","that","are","compatible","with","the","parent","class",",","so","that","`","from_config","`","can","be","used","from","a","class","different","than","the","one","used","to","save","the","config","(","should","be","overridden","by","parent","class",")","."]
docstr: Save a scheduler configuration object to the directory `save_directory`, so that it can be re-loaded using the
  [`~FlaxSchedulerMixin.from_pretrained`] class method.
  
  Args:
      save_directory (`str` or `os.PathLike`):
          Directory where the configuration JSON file will be saved (will be created if it does not exist).
      push_to_hub (`bool`, *optional*, defaults to `False`):
          Whether or not to push your model to the Hugging Face Hub after saving it. You can specify the
          repository you want to push to with `repo_id` (will default to the name of `save_directory` in your
          namespace).
      kwargs (`Dict[str, Any]`, *optional*):
          Additional keyword arguments passed along to the [`~utils.PushToHubMixin.push_to_hub`] method.
docstr_tok: ["Save","a","scheduler","configuration","object","to","the","directory","`","save_directory","`",",","so","that","it","can","be","re-loaded","using","the","[","`","~FlaxSchedulerMixin.from_pretrained","`","]","class","method",".","Args",":","save_directory","(","`","str","`","or","`","os.PathLike","`",")",":","Directory","where","the","configuration","JSON","file","will","be","saved","(","will","be","created","if","it","does","not","exist",")",".","push_to_hub","(","`","bool","`",",","*","optional","*",",","defaults","to","`","False","`",")",":","Whether","or","not","to","push","your","model","to","the","Hugging","Face","Hub","after","saving","it",".","You","can","specify","the","repository","you","want","to","push","to","with","`","repo_id","`","(","will","default","to","the","name","of","`","save_directory","`","in","your","namespace",")",".","kwargs","(","`","Dict","[","str",",","Any","]","`",",","*","optional","*",")",":","Additional","keyword","arguments","passed","along","to","the","[","`","~utils.PushToHubMixin.push_to_hub","`","]","method","."]
code: def save_pretrained(
      self,
      save_directory: Union[str, os.PathLike],
      push_to_hub: bool = False,
      **kwargs,
  ):
      """
      Save a scheduler configuration object to the directory `save_directory`, so that it can be re-loaded using the
      [`~FlaxSchedulerMixin.from_pretrained`] class method.
  
      Args:
          save_directory (`str` or `os.PathLike`):
              Directory where the configuration JSON file will be saved (will be created if it does not exist).
          push_to_hub (`bool`, *optional*, defaults to `False`):
              Whether or not to push your model to the Hugging Face Hub after saving it. You can specify the
              repository you want to push to with `repo_id` (will default to the name of `save_directory` in your
              namespace).
          kwargs (`Dict[str, Any]`, *optional*):
              Additional keyword arguments passed along to the [`~utils.PushToHubMixin.push_to_hub`] method.
      """
      self.save_config(
          save_directory=save_directory,
          push_to_hub=push_to_hub,
          **kwargs,
      )
  
imports: import importlib
  import math
  import os
  from dataclasses import dataclass
  from enum import Enum
  from typing import Optional, Tuple, Union
  import flax
  import jax.numpy
  from huggingface_hub.utils import validate_hf_hub_args
  from ..utils import BaseOutput, PushToHubMixin
label_desc: Use image node_id 2 for calling the FlaxSchedulerMixin obj's underlying member method code with example usage: obj.save_pretrained(save_directory, push_to_hub) without return types