Update generate_model_grid.py
Browse files- generate_model_grid.py +3 -0
generate_model_grid.py
CHANGED
@@ -244,6 +244,7 @@ class Script(scripts.Script):
|
|
244 |
for path in os.listdir(os.path.join(models_path, 'Stable-diffusion')):
|
245 |
if path.endswith('.ckpt'):
|
246 |
filenames.append(path)
|
|
|
247 |
|
248 |
with gr.Row():
|
249 |
x_values = gr.Textbox(label="Prompts, separated with &", lines=1)
|
@@ -264,6 +265,8 @@ class Script(scripts.Script):
|
|
264 |
if not no_fixed_seeds:
|
265 |
modules.processing.fix_seed(p)
|
266 |
|
|
|
|
|
267 |
if not opts.return_grid:
|
268 |
p.batch_size = 1
|
269 |
|
|
|
244 |
for path in os.listdir(os.path.join(models_path, 'Stable-diffusion')):
|
245 |
if path.endswith('.ckpt'):
|
246 |
filenames.append(path)
|
247 |
+
filenames.append('model.ckpt')
|
248 |
|
249 |
with gr.Row():
|
250 |
x_values = gr.Textbox(label="Prompts, separated with &", lines=1)
|
|
|
265 |
if not no_fixed_seeds:
|
266 |
modules.processing.fix_seed(p)
|
267 |
|
268 |
+
y_values = ','.join(y_values)
|
269 |
+
|
270 |
if not opts.return_grid:
|
271 |
p.batch_size = 1
|
272 |
|