monster/Qwen-Image-Layered-MRP-MLX
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

raise error if layer cannot be loaded

Browse Source 
Co-authored-by: Fabio Peruzzo <39661245+Xuzzo@users.noreply.github.com>
This commit is contained in:
Filip Strand 2024-09-04 19:43:32 +02:00 committed by GitHub
parent 0fd03c7552
commit b4c27a6fab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/flux_1/weights/weight_handler.py
View File

@ -86,7 +86,7 @@ class WeightHandler:
visited[parentKey]={} visited[parentKey]={}
visited[parentKey][splitKey]=weight visited[parentKey][splitKey]=weight
if not 'weight' in target: if not 'weight' in target:
continue raise ValueError(f"LoRA weights for layer {parentKey} cannot be loaded into the model.")
if 'lora_A' in visited[parentKey] and 'lora_B' in visited[parentKey]: if 'lora_A' in visited[parentKey] and 'lora_B' in visited[parentKey]:
lora_a=visited[parentKey]['lora_A'] lora_a=visited[parentKey]['lora_A']
lora_b=visited[parentKey]['lora_B'] lora_b=visited[parentKey]['lora_B']