Skip to content

Commit dc0bd7d

Browse files
committed
Change the parsing logic due to the api changes
1 parent 9519b1b commit dc0bd7d

File tree

1 file changed

+7
-8
lines changed

1 file changed

+7
-8
lines changed

src/together/resources/finetune.py

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff 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(

0 commit comments

Comments
 (0)