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

Remove restriction on lora weights

Browse Source
This commit is contained in:
filipstrand 2024-09-13 21:50:48 +02:00
parent d10e6f4019
commit db3e495be6
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/mflux/weights/lora_util.py
View File

@ -28,9 +28,6 @@ class LoraUtil:
@staticmethod @staticmethod
def _apply_lora(transformer: dict, lora_file: str, lora_scale: float) -> None: def _apply_lora(transformer: dict, lora_file: str, lora_scale: float) -> None:
if lora_scale < 0.0 or lora_scale > 1.0:
raise Exception(f"Invalid scale {lora_scale} provided for {lora_file}. Valid Range [0.0 - 1.0] ")
from mflux.weights.weight_handler import WeightHandler from mflux.weights.weight_handler import WeightHandler
lora_transformer, _ = WeightHandler.load_transformer(lora_path=lora_file) lora_transformer, _ = WeightHandler.load_transformer(lora_path=lora_file)
LoraUtil._apply_transformer(transformer, lora_transformer, lora_scale) LoraUtil._apply_transformer(transformer, lora_transformer, lora_scale)