JaehaerysNL/StableDiffusion
1
0
Fork 0
mirror of https://github.com/Stability-AI/stablediffusion.git synced 2024-12-22 15:44:58 +00:00
Code Issues Projects Releases Packages Wiki Activity

remove wrong arguments while using from_file.

Browse source
This commit is contained in:
Mohan-Zhang-u 2022-11-27 03:16:49 -05:00
parent 47b6b607fd
commit 941b9f9e97
1 changed files with 1 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
scripts/txt2img.py
View file

@ -165,12 +165,6 @@ def parse_args():
choices=["full", "autocast"],
default="autocast"
)
parser.add_argument(
"--repeat",
type=int,
default=1,
help="repeat each prompt in file this often",
)
opt = parser.parse_args()
return opt
@ -218,7 +212,7 @@ def main(opt):
print(f"reading prompts from {opt.from_file}")
with open(opt.from_file, "r") as f:
data = f.read().splitlines()
data = [p for p in data for i in range(opt.repeat)]
data = [p for p in data for i in range(batch_size)]
data = list(chunk(data, batch_size))
sample_path = os.path.join(outpath, "samples")