aboutsummaryrefslogtreecommitdiff
path: root/extensions-builtin
diff options
context:
space:
mode:
Diffstat (limited to 'extensions-builtin')
-rw-r--r--extensions-builtin/Lora/lora.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/extensions-builtin/Lora/lora.py b/extensions-builtin/Lora/lora.py
index 7a3ad9a9..6d860224 100644
--- a/extensions-builtin/Lora/lora.py
+++ b/extensions-builtin/Lora/lora.py
@@ -179,7 +179,10 @@ def list_available_loras():
os.makedirs(lora_dir, exist_ok=True)
- candidates = glob.glob(os.path.join(lora_dir, '**/*.pt'), recursive=True) + glob.glob(os.path.join(lora_dir, '**/*.safetensors'), recursive=True)
+ candidates = \
+ glob.glob(os.path.join(lora_dir, '**/*.pt'), recursive=True) + \
+ glob.glob(os.path.join(lora_dir, '**/*.safetensors'), recursive=True) + \
+ glob.glob(os.path.join(lora_dir, '**/*.ckpt'), recursive=True)
for filename in sorted(candidates):
if os.path.isdir(filename):
@@ -195,4 +198,3 @@ available_loras = {}
loaded_loras = []
list_available_loras()
-