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-04-24 17:03:09 +00:00
parent b078e4f936
commit 05892cf287
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tensors/cli.py
+10 -8
View File
@@ -794,7 +794,9 @@ def generate( # noqa: PLR0915
lora_strength: Annotated[float, typer.Option("--lora-strength", help="LoRA strength")] = 0.8, lora_strength: Annotated[float, typer.Option("--lora-strength", help="LoRA strength")] = 0.8,
negative: Annotated[str, typer.Option("-n", "--negative-prompt", help="Negative prompt")] = "", negative: Annotated[str, typer.Option("-n", "--negative-prompt", help="Negative prompt")] = "",
count: Annotated[int, typer.Option("-c", "--count", help="Number of images to generate")] = 1, count: Annotated[int, typer.Option("-c", "--count", help="Number of images to generate")] = 1,
rating: Annotated[str | None, typer.Option("--rating", "-R", help="Content rating: safe, questionable, explicit (Pony/Illustrious)")] = None, rating: Annotated[
str | None, typer.Option("--rating", "-R", help="Content rating: safe, questionable, explicit (Pony/Illustrious)")
] = None,
no_quality: Annotated[bool, typer.Option("--no-quality", help="Disable auto quality tags")] = False, no_quality: Annotated[bool, typer.Option("--no-quality", help="Disable auto quality tags")] = False,
no_negative: Annotated[bool, typer.Option("--no-negative", help="Disable auto negative prompt")] = False, no_negative: Annotated[bool, typer.Option("--no-negative", help="Disable auto negative prompt")] = False,
output: Annotated[Path | None, typer.Option("-o", "--output", help="Save path (default: current dir)")] = None, output: Annotated[Path | None, typer.Option("-o", "--output", help="Save path (default: current dir)")] = None,
@@ -1120,11 +1122,7 @@ def generate( # noqa: PLR0915
if output: if output:
img_data = get_image(str(img_path)) img_data = get_image(str(img_path))
if img_data: if img_data:
save_path = ( save_path = output if count == 1 else output.parent / f"{output.stem}_{i + 1:03d}{output.suffix}"
output
if count == 1
else output.parent / f"{output.stem}_{i + 1:03d}{output.suffix}"
)
save_path.write_bytes(img_data) save_path.write_bytes(img_data)
saved_path = save_path saved_path = save_path
all_saved.append(save_path) all_saved.append(save_path)
@@ -1482,8 +1480,12 @@ def db_cache(
@db_app.command("list") @db_app.command("list")
def db_list( def db_list(
model_type: Annotated[str | None, typer.Option("-t", "--type", help="Filter by model type (Checkpoint, LORA, VAE, etc.)")] = None, model_type: Annotated[
base: Annotated[str | None, typer.Option("-b", "--base", help="Filter by base model (Pony, Illustrious, SDXL 1.0, SD 1.5, etc.)")] = None, str | None, typer.Option("-t", "--type", help="Filter by model type (Checkpoint, LORA, VAE, etc.)")
] = None,
base: Annotated[
str | None, typer.Option("-b", "--base", help="Filter by base model (Pony, Illustrious, SDXL 1.0, SD 1.5, etc.)")
] = None,
json_output: Annotated[bool, typer.Option("--json", "-j", help="Output as JSON")] = False, json_output: Annotated[bool, typer.Option("--json", "-j", help="Output as JSON")] = False,
) -> None: ) -> None:
"""List local files with CivitAI info. """List local files with CivitAI info.