rename fp8 optimization in code for clarity as it does more than that with loras

2025-08-11 14:32:57 +03:00
parent 341932b58f
commit 7c93aea182
5 changed files with 26 additions and 21 deletions
@@ -1037,7 +1037,7 @@ class WanVideoModelLoader:
                    if k.endswith(".scale_weight"):
                        scale_weights[k] = v
            if not merge_loras:
-                from .fp8_optimization_v2 import _replace_linear
+                from .custom_linear import _replace_linear
                transformer = _replace_linear(transformer, base_dtype, sd, scale_weights=scale_weights)
                
            if "fp8_e4m3fn" in quantization: