Spaces:
Runtime error
Runtime error
Vivien Chappelier
commited on
Commit
Β·
3f4f0fe
1
Parent(s):
ff54256
add external script to run demo on a folder
Browse files- gen_sdxl_turbo.py +26 -0
gen_sdxl_turbo.py
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from app import BZHStableSignatureDemo
|
| 2 |
+
import sys
|
| 3 |
+
from pathlib import Path
|
| 4 |
+
|
| 5 |
+
def generate_all(captions, outdir='out'):
|
| 6 |
+
backend = BZHStableSignatureDemo()
|
| 7 |
+
decoders = list(backend.decoders.keys())
|
| 8 |
+
outdir = Path(outdir)
|
| 9 |
+
outdir.mkdir(parents=True, exist_ok=True)
|
| 10 |
+
|
| 11 |
+
for decoder in decoders:
|
| 12 |
+
# TODO: optimize by patching only once..
|
| 13 |
+
# TEMP
|
| 14 |
+
if decoder != 'medium':
|
| 15 |
+
continue
|
| 16 |
+
outdirdec = Path(outdir) / Path(decoder)
|
| 17 |
+
outdirdec.mkdir(parents=True, exist_ok = True)
|
| 18 |
+
for i, prompt in enumerate(captions):
|
| 19 |
+
img = backend.generate(decoder, 0, prompt)
|
| 20 |
+
img.save(outdirdec / Path('%04d.png' % i), format='PNG')
|
| 21 |
+
|
| 22 |
+
if __name__ == '__main__':
|
| 23 |
+
infile = sys.argv[1]
|
| 24 |
+
outdir = sys.argv[2]
|
| 25 |
+
with open(infile) as f:
|
| 26 |
+
generate_all([line.strip() for line in f.readlines()], outdir=outdir)
|