nodes/kijai-ComfyUI-WanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix fp8_fast when not using loras

Browse Source
This commit is contained in:
kijai
2025-09-08 18:54:29 +03:00
parent fe379fa77f
commit e5ef9752a7
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fp8_optimization.py
+1 -2
View File
@@ -56,8 +56,7 @@ def convert_fp8_linear(module, base_dtype, params_to_keep={}, scale_weight_keys=
if scale_weight_keys is not None:
scale_key = f"{name}.scale_weight"
if scale_key in scale_weight_keys:
print("Setting scale_weight for", name)
setattr(submodule, "scale_weight", scale_weight_keys[scale_key])
setattr(submodule, "scale_weight", scale_weight_keys[scale_key].float())
original_forward = submodule.forward
setattr(submodule, "original_forward", original_forward)
setattr(submodule, "forward", lambda input, m=submodule: fp8_linear_forward(m, base_dtype, input))