aladac/tensors
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

format: auto-format code [skip ci]

Browse Source
This commit is contained in:
github-actions[bot]
2026-05-18 19:05:18 +00:00
parent aa25d31ca9
commit 3f319bc016
1 changed files with 6 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tensors/cli.py
+6 -18
View File
@@ -2922,9 +2922,7 @@ app.add_typer(templates_app)
@templates_app.command("extract") @templates_app.command("extract")
def templates_extract( def templates_extract(
model: Annotated[str, typer.Argument(help="Local model name (e.g. lust_v10.safetensors)")], model: Annotated[str, typer.Argument(help="Local model name (e.g. lust_v10.safetensors)")],
orientation: Annotated[ orientation: Annotated[str, typer.Option("-O", "--orientation", help="Resolution: square, portrait, landscape")] = "portrait",
str, typer.Option("-O", "--orientation", help="Resolution: square, portrait, landscape")
] = "portrait",
no_overrides: Annotated[ no_overrides: Annotated[
bool, bool,
typer.Option( typer.Option(
@@ -2933,25 +2931,17 @@ def templates_extract(
), ),
] = False, ] = False,
api_key: Annotated[str | None, typer.Option("--api-key", help="CivitAI API key")] = None, api_key: Annotated[str | None, typer.Option("--api-key", help="CivitAI API key")] = None,
limit: Annotated[ limit: Annotated[int, typer.Option("--limit", "-L", help="Max templates to write (0 = all unique prompts)")] = 0,
int, typer.Option("--limit", "-L", help="Max templates to write (0 = all unique prompts)") overwrite: Annotated[bool, typer.Option("--overwrite", help="Overwrite existing template files (default: skip)")] = False,
] = 0,
overwrite: Annotated[
bool, typer.Option("--overwrite", help="Overwrite existing template files (default: skip)")
] = False,
do_generate: Annotated[ do_generate: Annotated[
bool, bool,
typer.Option("--generate", help="After writing, run `tsr generate --input` for each emitted template"), typer.Option("--generate", help="After writing, run `tsr generate --input` for each emitted template"),
] = False, ] = False,
output_dir: Annotated[ output_dir: Annotated[
Path | None, Path | None,
typer.Option( typer.Option("--output-dir", help="Where to write generated images when --generate (default: ComfyUI output dir)"),
"--output-dir", help="Where to write generated images when --generate (default: ComfyUI output dir)"
),
] = None, ] = None,
dry_run: Annotated[ dry_run: Annotated[bool, typer.Option("--dry-run", help="Print what would be done; write nothing")] = False,
bool, typer.Option("--dry-run", help="Print what would be done; write nothing")
] = False,
) -> None: ) -> None:
"""Bulk-extract templates from a model's CivitAI showcase. """Bulk-extract templates from a model's CivitAI showcase.
@@ -3113,9 +3103,7 @@ def templates_list(
items = list_templates(model) items = list_templates(model)
if json_output: if json_output:
console.print_json( console.print_json(data={"dir": str(TEMPLATES_DIR), "templates": [{"model": m, "name": n} for m, n in items]})
data={"dir": str(TEMPLATES_DIR), "templates": [{"model": m, "name": n} for m, n in items]}
)
return return
if not items: if not items:
scope = f" for model '{model}'" if model else "" scope = f" for model '{model}'" if model else ""