black

2 years ago · f7555a3e3d
parent 94f4f25e9b
commit f7555a3e3d
1 changed files with 1 additions and 3 deletions
--- a/src/utils/convert_8bit.py
+++ b/src/utils/convert_8bit.py
@ -31,8 +31,6 @@ def replace_8bit_linear(model, threshold=6.0):
                threshold=threshold,
            )
            model._modules[n].weight = bnb.nn.Int8Params(
-                module.weight.data, 
-                requires_grad=False,
-                has_fp16_weights=False
+                module.weight.data, requires_grad=False, has_fp16_weights=False
            ).to(module.weight.dtype)
    return model