skygpu
/
skynet
mirror of https://github.com/skygpu/skynet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update typo in cli call to txt2txt util

txt2txt
Konstantine Tsafatinos 2025-01-09 21:27:26 -05:00
parent 4dce4ee9e3
commit 76b0e99874
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
skynet/cli.py
View File

@ -106,7 +106,7 @@ def inpaint(model, prompt, input, mask, output, strength, guidance, steps, seed)
@click.option('--temperature', '-t', default=1.0) @click.option('--temperature', '-t', default=1.0)
@click.option('--max-length', '-ml', default=256) @click.option('--max-length', '-ml', default=256)
@click.option('--num-return-sequences', '-rs', default=1) @click.option('--num-return-sequences', '-rs', default=1)
@click.option('--no-repeat-ngram', '-nr', default=2) @click.option('--no-repeat-ngram_size', '-nr', default=2)
@click.option('--top-p', '-tp', default=0.95) @click.option('--top-p', '-tp', default=0.95)
def txt2txt( def txt2txt(
model, model,
@ -115,7 +115,7 @@ def txt2txt(
temperature, temperature,
max_length, max_length,
num_return_sequences, num_return_sequences,
no_repeat_ngram, no_repeat_ngram_size,
top_p top_p
): ):
from . import utils from . import utils
@ -131,7 +131,7 @@ def txt2txt(
temperature=temperature, temperature=temperature,
max_length=max_length, max_length=max_length,
num_return_sequences=num_return_sequences, num_return_sequences=num_return_sequences,
no_repeat_ngram=no_repeat_ngram, no_repeat_ngram_size=no_repeat_ngram_size,
top_p=top_p top_p=top_p
) )