Froddan commited on
Commit
365ec21
1 Parent(s): d9ef16e

Update generate_model_grid.py

Browse files
Files changed (1) hide show
  1. 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