Commit 659d602d authored by EllangoK's avatar EllangoK
Browse files

only returns ckpt directories if they are not none

parent 0a851508

modules/ui_extra_networks_checkpoints.py

+1 −1
Original line number Diff line number Diff line
@@ -34,5 +34,5 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage): 
            }



    def allowed_directories_for_previews(self):

        return [shared.cmd_opts.ckpt_dir, sd_models.model_path]

        return [v for v in [shared.cmd_opts.ckpt_dir, sd_models.model_path] if v is not None]

CRA Git | Maintained and supported by SUSTech CRA and CCSE