Fix unmerged lora application on unquantized models

This commit is contained in:
kijai
2025-08-11 01:34:25 +03:00
parent d6270ace0a
commit 341932b58f
+3 -3
View File
@@ -1036,9 +1036,9 @@ class WanVideoModelLoader:
for k, v in sd.items():
if k.endswith(".scale_weight"):
scale_weights[k] = v
if not merge_loras:
from .fp8_optimization_v2 import _replace_linear
transformer = _replace_linear(transformer, base_dtype, sd, scale_weights=scale_weights)
if not merge_loras:
from .fp8_optimization_v2 import _replace_linear
transformer = _replace_linear(transformer, base_dtype, sd, scale_weights=scale_weights)
if "fp8_e4m3fn" in quantization:
dtype = torch.float8_e4m3fn