File tree Expand file tree Collapse file tree 1 file changed +2
-12
lines changed
Expand file tree Collapse file tree 1 file changed +2
-12
lines changed Original file line number Diff line number Diff line change @@ -235,22 +235,12 @@ def _parse_raw_checkpoints(
235235 Returns:
236236 List[FinetuneCheckpoint]: List of available checkpoints
237237 """
238- has_adapters = any (ckpt ["is_adapter" ] for ckpt in checkpoints )
239238
240239 parsed_checkpoints = []
241240 for checkpoint in checkpoints :
242241 step = checkpoint ["step" ]
243- is_adapter = checkpoint ["is_adapter" ]
244- is_final = checkpoint ["is_final" ]
245- checkpoint_name = f"{ id } :{ step } " if not is_final else id
246-
247- if is_final :
248- if is_adapter :
249- checkpoint_type = "Final Adapter"
250- else :
251- checkpoint_type = "Final Merged" if has_adapters else "Final"
252- else :
253- checkpoint_type = f"Intermediate (step { step } )"
242+ checkpoint_type = checkpoint ["checkpoint_type" ]
243+ checkpoint_name = f"{ id } :{ step } " if "intermediate" in checkpoint_type .lower () else id
254244
255245 parsed_checkpoints .append (
256246 FinetuneCheckpoint (
You can’t perform that action at this time.
0 commit comments