From 829c4550f74c2fd6cdc5d95e0600c6436bbf1bca Mon Sep 17 00:00:00 2001 From: Daniel Dale Date: Wed, 20 Dec 2023 08:22:08 -0800 Subject: [PATCH] adjust decorator order to accommodate older versions of python --- src/finetuning_scheduler/fts_supporters.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/finetuning_scheduler/fts_supporters.py b/src/finetuning_scheduler/fts_supporters.py index 1a65c9f..ae1de8c 100644 --- a/src/finetuning_scheduler/fts_supporters.py +++ b/src/finetuning_scheduler/fts_supporters.py @@ -1362,8 +1362,8 @@ def gen_implicit_schedule(self, sched_dir: Union[str, os.PathLike]) -> None: rank_zero_info(f"Generated default fine-tuning schedule '{default_ft_schedule}' for iterative fine-tuning") self.ft_schedule = self.load_yaml_schedule(default_ft_schedule) - @rank_zero_only @staticmethod + @rank_zero_only def save_schedule(schedule_name: str, layer_config: Dict, dump_loc: Union[str, os.PathLike]) -> os.PathLike: """Save loaded or generated schedule to a directory to ensure reproducability. @@ -1387,8 +1387,8 @@ def save_schedule(schedule_name: str, layer_config: Dict, dump_loc: Union[str, o rank_zero_info(f"fine-tuning schedule dumped to {ft_schedule_yaml}.") return ft_schedule_yaml - @rank_zero_only @staticmethod + @rank_zero_only def gen_ft_schedule(module: Module, dump_loc: Union[str, os.PathLike]) -> Optional[os.PathLike]: """Generate the default fine-tuning schedule using a naive, 2-parameters per-level heuristic.