Skip to content

FIX MiSS with bat init error with some quantization backends#3381

Open
BenjaminBossan wants to merge 1 commit into
huggingface:mainfrom
BenjaminBossan:fix-miss-bat-init-and-quantization-support
Open

FIX MiSS with bat init error with some quantization backends#3381
BenjaminBossan wants to merge 1 commit into
huggingface:mainfrom
BenjaminBossan:fix-miss-bat-init-and-quantization-support

Conversation

@BenjaminBossan

Copy link
Copy Markdown
Member

When using MiSS with the init_weights='bat', the forward pass requires getting the non-quantized base weights. When the quantization backend does not support dequantization, we now detect this and raise a helpful error message. Tests have been updated to check for this error message.

Normally, we should have caught this error. The reason why we didn't is that we have a helper function for testing that changes PEFT config kwargs to be non-zero initialization. When passing init_weights='bat' to that function, we would override 'bat' with False. Thus, the 'bat' path was not actually being tested. The helper function has now been updated not to override 'bat'.

When using MiSS with the init_weights='bat', the forward pass requires
getting the non-quantized base weights. When the quantization backend
does not support dequantization, we now detect this and raise a
helpful error message. Tests have been updated to check for this error
message.

Normally, we should have caught this error. The reason why we didn't
is that we have a helper function for testing that changes PEFT config
kwargs to be non-zero initialization. When passing init_weights='bat'
to that function, we would override 'bat' with False. Thus, the 'bat'
path was not actually being tested. The helper function has now been
updated not to override 'bat'.
@HuggingFaceDocBuilderDev

Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@BenjaminBossan BenjaminBossan requested a review from githubnemo July 1, 2026 17:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants