Skip to content

Commit 5522954

Browse files
committed
fix: reject hopper for nunchaku lite
1 parent 461367a commit 5522954

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

src/diffusers/quantizers/nunchaku/nunchaku_quantizer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,9 @@ def validate_environment(self, *args, **kwargs):
3535

3636
device_capability = torch.cuda.get_device_capability()
3737

38+
if device_capability[0] == 9:
39+
raise ValueError("Loading Nunchaku checkpoints is not supported on Hopper NVIDIA GPUs.")
40+
3841
has_nvfp4_config = (
3942
self.quantization_config.svdq_w4a4 is not None
4043
and self.quantization_config.svdq_w4a4["precision"] == "nvfp4"

0 commit comments

Comments
 (0)