We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0fc321e commit 4b6075cCopy full SHA for 4b6075c
src/lightning/fabric/plugins/precision/bitsandbytes.py
@@ -15,11 +15,14 @@
15
from typing import Any, Literal, Optional
16
17
import torch
18
+from lightning_utilities.core.imports import RequirementCache
19
from typing_extensions import override
20
21
from lightning.fabric.plugins.precision.precision import Precision
22
from lightning.fabric.utilities.imports import _raise_enterprise_not_available
23
24
+_BITSANDBYTES_AVAILABLE = RequirementCache("bitsandbytes")
25
+
26
27
class BitsandbytesPrecision(Precision):
28
"""Plugin for quantizing weights with `bitsandbytes <https://github.com/bitsandbytes-foundation/bitsandbytes>`__.
0 commit comments