test comment removed

45864262 · Ruslan Svirschevski · de44be1a · 45864262
Commit 45864262 authored Nov 10, 2023 by Ruslan Svirschevski
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_linear4bit.py tests/test_linear4bit.py +1 -1

No files found.
--- a/tests/test_linear4bit.py
+++ b/tests/test_linear4bit.py
@@ -30,7 +30,7 @@ def test_linear_serialization(quant_type, compress_statistics, bias):
        compute_dtype=compute_dtype,
        compress_statistics=compress_statistics,
        quant_type=quant_type,
-        device="meta",  # TODO: consider both CPU, meta and CUDA creation
+        device="meta",
    )
    new_weight = bnb.nn.Params4bit(data=linear.weight, requires_grad=False)
    linear_q.weight = new_weight