unsloth/cli/__init__.py

19 lines
546 B
Python

import typer
from cli.commands.train import train
from cli.commands.inference import inference
from cli.commands.export import export, list_checkpoints
from cli.commands.ui import ui
from cli.commands.studio import studio
app = typer.Typer(
help="Command-line interface for Unsloth training, inference, and export.",
context_settings={"help_option_names": ["-h", "--help"]},
)
app.command()(train)
app.command()(inference)
app.command()(export)
app.command("list-checkpoints")(list_checkpoints)
app.command()(ui)
app.command()(studio)