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