Update generate_model_grid.py
Browse files- generate_model_grid.py +3 -3
generate_model_grid.py
CHANGED
@@ -241,11 +241,12 @@ class Script(scripts.Script):
|
|
241 |
|
242 |
def ui(self, is_img2img):
|
243 |
filenames = []
|
|
|
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)
|
251 |
|
@@ -262,11 +263,10 @@ class Script(scripts.Script):
|
|
262 |
return [x_values, y_values, z_values, draw_legend, include_lone_images, no_fixed_seeds]
|
263 |
|
264 |
def run(self, p, x_values, y_values, z_values, draw_legend, include_lone_images, no_fixed_seeds):
|
|
|
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 |
|
|
|
241 |
|
242 |
def ui(self, is_img2img):
|
243 |
filenames = []
|
244 |
+
z_valuez = ''
|
245 |
for path in os.listdir(os.path.join(models_path, 'Stable-diffusion')):
|
246 |
if path.endswith('.ckpt'):
|
247 |
filenames.append(path)
|
248 |
filenames.append('model.ckpt')
|
249 |
+
|
250 |
with gr.Row():
|
251 |
x_values = gr.Textbox(label="Prompts, separated with &", lines=1)
|
252 |
|
|
|
263 |
return [x_values, y_values, z_values, draw_legend, include_lone_images, no_fixed_seeds]
|
264 |
|
265 |
def run(self, p, x_values, y_values, z_values, draw_legend, include_lone_images, no_fixed_seeds):
|
266 |
+
y_values = ','.join(y_values)
|
267 |
if not no_fixed_seeds:
|
268 |
modules.processing.fix_seed(p)
|
269 |
|
|
|
|
|
270 |
if not opts.return_grid:
|
271 |
p.batch_size = 1
|
272 |
|