Skip to content

Commit 80f1d34

Browse files
committed
revert Fabric precision change: gives ValueError 'Choose one [precision type].'
1 parent 2eb1221 commit 80f1d34

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

thunder/tests/test_networks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -292,7 +292,7 @@ def test_quantization():
292292
import lightning as L
293293

294294
plugins = BitsandbytesPrecision("nf4", torch.bfloat16)
295-
fabric = L.Fabric(devices=1, precision="bf16-mixed", plugins=plugins)
295+
fabric = L.Fabric(devices=1, precision=None, plugins=plugins)
296296
with fabric.init_module(empty_init=True):
297297
model = litgpt_model.GPT(config)
298298

0 commit comments

Comments
 (0)