Change batch defaults to be friendlier to lower Vram cards

This commit is contained in:
kjerk 2022-11-24 01:53:52 -08:00
parent 33910c386e
commit cec55255ca
2 changed files with 3 additions and 3 deletions

View file

@ -126,7 +126,7 @@ def main():
parser.add_argument(
"--n_samples",
type=int,
default=2,
default=1,
help="how many samples to produce for each given prompt. A.k.a batch size",
)

View file

@ -91,7 +91,7 @@ def parse_args():
parser.add_argument(
"--n_iter",
type=int,
default=3,
default=2,
help="sample this often",
)
parser.add_argument(
@ -121,7 +121,7 @@ def parse_args():
parser.add_argument(
"--n_samples",
type=int,
default=3,
default=1,
help="how many samples to produce for each given prompt. A.k.a batch size",
)
parser.add_argument(