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 cfa846e commit b1ec22dCopy full SHA for b1ec22d
unsloth/models/llama.py
@@ -1825,7 +1825,7 @@ def from_pretrained(
1825
1826
# Convert to HF format
1827
_, quant_state_dict = get_vllm_state_dict(llm, config = model_config)
1828
- model = convert_vllm_to_huggingface(quant_state_dict, model_config, dtype)
+ model = convert_vllm_to_huggingface(quant_state_dict, model_config, dtype, bnb_config)
1829
model.vllm_engine = llm
1830
model.fast_generate = model.vllm_engine.generate
1831
model.fast_generate_batches = functools.partial(generate_batches, model.vllm_engine)
0 commit comments