6f77c63229
- Workers now compute backend_path and venv_t5 locally via Path(__file__) - Moved .venv_t5 to ~/.unsloth/studio/.venv_t5 - Added ensure_studio_directories() call on server startup - Expanded CLI studio command into sub-app with setup subcommand
22 lines
706 B
Python
22 lines
706 B
Python
# SPDX-License-Identifier: AGPL-3.0-only - See /studio/LICENSE.AGPL-3.0
|
|
# Copyright © 2025 Unsloth AI
|
|
|
|
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
|
|
|
|
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.add_typer(studio_app, name="studio", help="Unsloth Studio commands.")
|