Bit more multi-line styling in `.cli`

Such that i could more easily read the flag specs in a vsplit `vim`
buffer set ;)

Also includes the same for some (internal) `tuple`-imports.
fomo_polish
Tyler Goodlet 2025-02-03 10:46:09 -05:00
parent fd3e0feffe
commit e829c6c569
1 changed files with 39 additions and 8 deletions

View File

@ -8,10 +8,25 @@ from functools import partial
import click import click
from leap.protocol import Name, Asset from leap.protocol import (
Name,
Asset,
)
from .config import * from .config import (
from .constants import * load_skynet_toml,
load_key,
set_hf_vars,
ConfigParsingError,
)
from .constants import (
# TODO, more conventional to make these private i'm pretty
# sure according to pep8?
DEFAULT_IPFS_DOMAIN,
DEFAULT_EXPLORER_DOMAIN,
DEFAULT_CONFIG_PATH,
MODELS,
)
@click.group() @click.group()
@ -22,7 +37,10 @@ def skynet(*args, **kwargs):
@click.command() @click.command()
@click.option('--model', '-m', default=list(MODELS.keys())[-1]) @click.option('--model', '-m', default=list(MODELS.keys())[-1])
@click.option( @click.option(
'--prompt', '-p', default='a red old tractor in a sunny wheat field') '--prompt',
'-p',
default='a red old tractor in a sunny wheat field',
)
@click.option('--output', '-o', default='output.png') @click.option('--output', '-o', default='output.png')
@click.option('--width', '-w', default=512) @click.option('--width', '-w', default=512)
@click.option('--height', '-h', default=512) @click.option('--height', '-h', default=512)
@ -30,7 +48,7 @@ def skynet(*args, **kwargs):
@click.option('--steps', '-s', default=26) @click.option('--steps', '-s', default=26)
@click.option('--seed', '-S', default=None) @click.option('--seed', '-S', default=None)
def txt2img(*args, **kwargs): def txt2img(*args, **kwargs):
from . import utils from . import utils # TODO? why here, import cycle?
config = load_skynet_toml() config = load_skynet_toml()
hf_token = load_key(config, 'skynet.dgpu.hf_token') hf_token = load_key(config, 'skynet.dgpu.hf_token')
@ -38,10 +56,18 @@ def txt2img(*args, **kwargs):
set_hf_vars(hf_token, hf_home) set_hf_vars(hf_token, hf_home)
utils.txt2img(hf_token, **kwargs) utils.txt2img(hf_token, **kwargs)
@click.command() @click.command()
@click.option('--model', '-m', default=list(MODELS.keys())[-2])
@click.option( @click.option(
'--prompt', '-p', default='a red old tractor in a sunny wheat field') '--model',
'-m',
default=list(MODELS.keys())[-2]
)
@click.option(
'--prompt',
'-p',
default='a red old tractor in a sunny wheat field',
)
@click.option('--input', '-i', default='input.png') @click.option('--input', '-i', default='input.png')
@click.option('--output', '-o', default='output.png') @click.option('--output', '-o', default='output.png')
@click.option('--strength', '-Z', default=1.0) @click.option('--strength', '-Z', default=1.0)
@ -118,6 +144,7 @@ def download():
set_hf_vars(hf_token, hf_home) set_hf_vars(hf_token, hf_home)
utils.download_all_models(hf_token, hf_home) utils.download_all_models(hf_token, hf_home)
@skynet.command() @skynet.command()
@click.option( @click.option(
'--reward', '-r', default='20.0000 GPU') '--reward', '-r', default='20.0000 GPU')
@ -315,6 +342,7 @@ def config(
def deposit(quantity: str): def deposit(quantity: str):
import trio import trio
from leap.cleos import CLEOS from leap.cleos import CLEOS
from leap.sugar import asset_from_str
config = load_skynet_toml() config = load_skynet_toml()
@ -365,7 +393,10 @@ def nodeos():
@run.command() @run.command()
@click.option('--loglevel', '-l', default='INFO', help='Logging level') @click.option('--loglevel', '-l', default='INFO', help='Logging level')
@click.option( @click.option(
'--config-path', '-c', default=DEFAULT_CONFIG_PATH) '--config-path',
'-c',
default=DEFAULT_CONFIG_PATH,
)
def dgpu( def dgpu(
loglevel: str, loglevel: str,
config_path: str config_path: str