fix typo in python/sglang/srt/layers/quantization/fp8.py (#2655)

e6f523b5 · HAI · GitHub · 32318178 · e6f523b5
Unverified Commit e6f523b5 authored Dec 29, 2024 by HAI Committed by GitHub Dec 29, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

python/sglang/srt/layers/quantization/fp8.py python/sglang/srt/layers/quantization/fp8.py +2 -2

No files found.
--- a/python/sglang/srt/layers/quantization/fp8.py
+++ b/python/sglang/srt/layers/quantization/fp8.py
@@ -280,9 +280,9 @@ class Fp8LinearMethod(LinearMethodBase):
                    weight_scale=layer.weight_scale_inv,
                    input_scale=None,
                )
-                layer.weight = torch.nn.Parameter(weight, require_grad=False)
+                layer.weight = torch.nn.Parameter(weight, requires_grad=False)
                layer.weight_scale_inv = torch.nn.Parameter(
-                    weight_scale, require_grad=False
+                    weight_scale, requires_grad=False
                )
                layer.input_scale = None
            return