Update generate_model_grid.py

Should now also look in subfolders

generate_model_grid.py

@@ -242,9 +242,16 @@ class Script(scripts.Script):
     def ui(self, is_img2img):
         filenames = []
         z_valuez = ''
-        for path in os.listdir(os.path.join(models_path, 'Stable-diffusion')):
+        dirpath = os.path.join(models_path, 'Stable-diffusion')
+        for path in os.listdir(dirpath):
             if path.endswith('.ckpt'):
                 filenames.append(path)
+            else:
+                if os.path.isdir(os.path.join(dirpath,path)):
+                  for subpath in os.listdir(os.path.join(dirpath,path)):
+                    if subpath.endswith('.ckpt'):
+                      filenames.append(subpath)
+            
         filenames.append('model.ckpt')
         
         with gr.Row():