src/hyperpod_nemo_adapter/utils/callbacks/ckpt_io.py [55:62]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def save_checkpoint(
        self,
        checkpoint: Dict[str, Any],
        path: _PATH,
        storage_options: Optional[Any] = None,
        modeling_class: Optional[Any] = AutoModelForCausalLM,
        model_config: Optional[Any] = None,
    ) -> None:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py [47:54]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def save_checkpoint(
        self,
        checkpoint: Dict[str, Any],
        path: _PATH,
        storage_options: Optional[Any] = None,
        modeling_class: Optional[Any] = AutoModelForCausalLM,
        model_config: Optional[Any] = None,
    ) -> None:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -