Skip to content

Commit d49623a

Browse files
committed
Parsing updated
1 parent 892b1ee commit d49623a

File tree

1 file changed

+2
-12
lines changed

1 file changed

+2
-12
lines changed

src/together/resources/finetune.py

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

0 commit comments

Comments
 (0)