Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

improve support for manually downloaded controlnets

Browse Source
This commit is contained in:
Vladimir Mandic
2024-04-07 12:03:18 -04:00
parent b1e6a81bef
commit 3ddf5751f5
4 changed files with 33 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/control/units/controlnet.py
+5
View File
@@ -69,14 +69,19 @@ cache_dir = 'models/control/controlnet'
def find_models():
path = os.path.join(opts.control_dir, 'controlnet')
files = listdir(path)
folders = [f for f in files if os.path.isdir(f) if os.path.exists(os.path.join(f, 'config.json'))]
files = [f for f in files if f.endswith('.safetensors')]
downloaded_models = {}
for f in files:
basename = os.path.splitext(os.path.relpath(f, path))[0]
downloaded_models[basename] = os.path.join(path, f)
for f in folders:
basename = os.path.relpath(f, path)
downloaded_models[basename] = f
all_models.update(downloaded_models)
return downloaded_models
find_models()
def list_models(refresh=False):
import modules.shared