Skip to content

Commit 6365a7c

Browse files
authored
Merge branch 'v2.3' into lstein/bugfix/improve-update-handling
2 parents 5fcb3d9 + d81584c commit 6365a7c

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

ldm/invoke/_version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__='2.3.4'
1+
__version__='2.3.4.post1'

ldm/modules/kohya_lora_manager.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -339,12 +339,14 @@ def load_lora_module(self, name, path_file, multiplier: float = 1.0):
339339
return lora
340340

341341
def apply_lora_model(self, name, mult: float = 1.0):
342+
path_file = None
342343
for suffix in ["ckpt", "safetensors", "pt"]:
343-
path_file = Path(self.lora_path, f"{name}.{suffix}")
344-
if path_file.exists():
344+
path_files = [x for x in Path(self.lora_path).glob(f"**/{name}.{suffix}")]
345+
if len(path_files):
346+
path_file = path_files[0]
345347
print(f" | Loading lora {path_file.name} with weight {mult}")
346348
break
347-
if not path_file.exists():
349+
if not path_file:
348350
print(f" ** Unable to find lora: {name}")
349351
return
350352

0 commit comments

Comments
 (0)