File tree Expand file tree Collapse file tree 1 file changed +7
-8
lines changed
Expand file tree Collapse file tree 1 file changed +7
-8
lines changed Original file line number Diff line number Diff line change @@ -235,23 +235,22 @@ def _parse_raw_checkpoints(
235235 Returns:
236236 List[FinetuneCheckpoint]: List of available checkpoints
237237 """
238- had_adapters = any (ckpt ["path" ]. endswith ( "_adapter" ) for ckpt in checkpoints )
238+ has_adapters = any (ckpt ["is_adapter" ] for ckpt in checkpoints )
239239
240240 parsed_checkpoints = []
241241 for checkpoint in checkpoints :
242- checkpoint_path = checkpoint ["path" ]
243242 step = checkpoint ["step" ]
244-
245- is_final = int ( step ) == 0
246- checkpoint_name = f"{ id } :step" if step else id
243+ is_adapter = checkpoint [ "is_adapter" ]
244+ is_final = checkpoint [ "is_final" ]
245+ checkpoint_name = f"{ id } :{ step } " if not is_final else id
247246
248247 if is_final :
249- if checkpoint_path . endswith ( "_adapter" ) :
248+ if is_adapter :
250249 checkpoint_type = "Final Adapter"
251250 else :
252- checkpoint_type = "Final Merged" if had_adapters else "Final"
251+ checkpoint_type = "Final Merged" if has_adapters else "Final"
253252 else :
254- checkpoint_type = "Intermediate"
253+ checkpoint_type = "Intermediate (step {step}) "
255254
256255 parsed_checkpoints .append (
257256 FinetuneCheckpoint (
You can’t perform that action at this time.
0 commit comments