CLI fix for backend changes
This commit is contained in:
parent
b430e23c0e
commit
b8aa528cf3
2 changed files with 11 additions and 6 deletions
|
|
@ -22,8 +22,11 @@ def list_checkpoints(
|
|||
typer.echo("No checkpoints found.")
|
||||
raise typer.Exit()
|
||||
|
||||
for display, path in checkpoints:
|
||||
typer.echo(f"{display}: {path}")
|
||||
for model_name, ckpt_list, metadata in checkpoints:
|
||||
typer.echo(f"\n{model_name}:")
|
||||
for display, path, loss in ckpt_list:
|
||||
loss_str = f" (loss: {loss:.4f})" if loss is not None else ""
|
||||
typer.echo(f" {display}{loss_str}: {path}")
|
||||
|
||||
|
||||
def export(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue