Downgrade ligthning to before bitsandbytes upgrade (#1104)

Author: carmocca

pyproject.toml

@@ -10,7 +10,7 @@ license = { file = "LICENSE" }
 
 dependencies = [
     "torch>=2.2.0",
-    "lightning @ git+https://github.com/Lightning-AI/lightning@f23b3b1e7fdab1d325f79f69a28706d33144f27e",
+    "lightning @ git+https://github.com/Lightning-AI/lightning@b19c3a961c79028d7c39a4f1ff1c2df991406d1d",
     # TODO: install from PyPI when https://github.com/omni-us/jsonargparse/pull/466 is released
     "jsonargparse[signatures] @ git+https://github.com/omni-us/jsonargparse",
 ]

tests/test_ci.py

@@ -1,19 +1,10 @@
 # Copyright Lightning AI. Licensed under the Apache License 2.0, see LICENSE file.
 
-# this file is just to validate on the CI logs that these tests were run
+from lightning.fabric.plugins.precision.bitsandbytes import _BITSANDBYTES_AVAILABLE
+
 from conftest import RunIf
 
 
 @RunIf(min_cuda_gpus=1)
-def test_runif_min_cuda_gpus():
-    assert True
-
-
-@RunIf(min_cuda_gpus=1, standalone=True)
-def test_runif_min_cuda_gpus_standalone():
-    assert True
-
-
-@RunIf(standalone=True)
-def test_runif_standalone():
-    assert True
+def test_gpu_ci_installs_bitsandbytes():
+    assert _BITSANDBYTES_AVAILABLE, str(_BITSANDBYTES_AVAILABLE)

tests/test_lora.py

@@ -689,8 +689,8 @@ def test_lora_bitsandbytes(monkeypatch, tmp_path, fake_checkpoint_dir, alpaca_pa
         },
     }
 
-    assert {p.name for p in tmp_path.rglob("*.pth")} == {"lit_model.pth"}
-    state_dict = torch.load(tmp_path / "final" / "lit_model.pth")
+    assert {p.name for p in tmp_path.rglob("*.lora")} == {"lit_model.pth.lora"}
+    state_dict = torch.load(tmp_path / "final" / "lit_model.pth.lora")
     assert len(state_dict) == 1
     dtype_to_name = {"torch.float16": set()}
     for name, layer in state_dict["model"].items():