From ca87c09c0ec9f6d3fbfda5d8579907c7017eb629 Mon Sep 17 00:00:00 2001
From: d8ahazard <d8ahazard@gmail.com>
Date: Fri, 30 Sep 2022 09:28:28 -0500
Subject: [PATCH] Fix recursive model loading

Ensure we find checkpoints within subdirectories.
---
 modules/modelloader.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/modules/modelloader.py b/modules/modelloader.py
index 1106aeb7f..df29b6ef6 100644
--- a/modules/modelloader.py
+++ b/modules/modelloader.py
@@ -1,3 +1,4 @@
+import glob
 import os
 import shutil
 import importlib
@@ -41,7 +42,7 @@ def load_models(model_path: str, model_url: str = None, command_path: str = None
 
         for place in places:
             if os.path.exists(place):
-                for file in os.listdir(place):
+                for file in glob.iglob(place + '**/**', recursive=True):
                     full_path = os.path.join(place, file)
                     if os.path.isdir(full_path):
                         continue
@@ -50,6 +51,7 @@ def load_models(model_path: str, model_url: str = None, command_path: str = None
                         if extension not in ext_filter:
                             continue
                     if file not in output:
+                        print(f"FILE: {full_path}")
                         output.append(full_path)
 
         if model_url is not None and len(output) == 0:
-- 
GitLab