You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
27 lines
523 B
Python
27 lines
523 B
Python
import click
|
|
|
|
from . import run_stable_vicuna
|
|
|
|
|
|
@click.group()
|
|
def cli():
|
|
pass
|
|
|
|
|
|
@cli.command()
|
|
@click.option(
|
|
"--model-dir",
|
|
type=click.Path(exists=True, file_okay=False, dir_okay=True),
|
|
envvar="SAVANT_MODEL_DIR",
|
|
default="~/models",
|
|
show_default=True,
|
|
help="The path to a directory containing a hugging face Stable-Vicuna model.",
|
|
)
|
|
def stable_vicuna(model_dir):
|
|
"""Runs a Stable Vicuna CLI prompt."""
|
|
run_stable_vicuna.main(model_dir=model_dir)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
cli()
|