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