HPU test update

Author: matthewdouglas

tests/test_modules.py

@@ -299,7 +299,7 @@ def test_kbit_backprop(device, module):
     kbit = nn.Sequential(*[torch.nn.Linear(dim1, dim2), module(dim2, 128)])
 
     if device == "hpu":
-        if isinstance(module, bnb.nn.LinearFP4):
+        if isinstance(kbit, bnb.nn.LinearFP4):
             pytest.skip("FP4 is not supported on HPU")
 
     kbit[0].weight.detach().copy_(ref[0].weight)