Skip to content

Commit 967d853

Browse files
authored
Merge branch 'v2.3' into feat/lokr_support
2 parents e91117b + d81584c commit 967d853

File tree

3 files changed

+7
-5
lines changed

3 files changed

+7
-5
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
@@ -427,12 +427,14 @@ def load_lora_module(self, name, path_file, multiplier: float = 1.0):
427427
return lora
428428

429429
def apply_lora_model(self, name, mult: float = 1.0):
430+
path_file = None
430431
for suffix in ["ckpt", "safetensors", "pt"]:
431-
path_file = Path(self.lora_path, f"{name}.{suffix}")
432-
if path_file.exists():
432+
path_files = [x for x in Path(self.lora_path).glob(f"**/{name}.{suffix}")]
433+
if len(path_files):
434+
path_file = path_files[0]
433435
print(f" | Loading lora {path_file.name} with weight {mult}")
434436
break
435-
if not path_file.exists():
437+
if not path_file:
436438
print(f" ** Unable to find lora: {name}")
437439
return
438440

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ dependencies = [
3434
"clip_anytorch",
3535
"compel~=1.1.0",
3636
"datasets",
37-
"diffusers[torch]~=0.14",
37+
"diffusers[torch]==0.14",
3838
"dnspython==2.2.1",
3939
"einops",
4040
"eventlet",

0 commit comments

Comments
 (0)