Skip to content

Commit 4b6075c

Browse files
committed
update
1 parent 0fc321e commit 4b6075c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/lightning/fabric/plugins/precision/bitsandbytes.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,14 @@
1515
from typing import Any, Literal, Optional
1616

1717
import torch
18+
from lightning_utilities.core.imports import RequirementCache
1819
from typing_extensions import override
1920

2021
from lightning.fabric.plugins.precision.precision import Precision
2122
from lightning.fabric.utilities.imports import _raise_enterprise_not_available
2223

24+
_BITSANDBYTES_AVAILABLE = RequirementCache("bitsandbytes")
25+
2326

2427
class BitsandbytesPrecision(Precision):
2528
"""Plugin for quantizing weights with `bitsandbytes <https://github.com/bitsandbytes-foundation/bitsandbytes>`__.

0 commit comments

Comments
 (0)