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 |
|