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 70755e8 commit ebd8c66Copy full SHA for ebd8c66
vllm/lora/layers.py
@@ -51,6 +51,9 @@ def _get_lora_device(base_layer: nn.Module) -> torch.device:
51
# marlin
52
elif hasattr(base_layer, "B"):
53
return base_layer.B.device
54
+ # HQQ marlin
55
+ elif hasattr(base_layer, "W_q"):
56
+ return base_layer.W_q.device
57
else:
58
raise ValueError(f"Unsupported base layer: {base_layer}")
59
0 commit comments