Skip to content

Commit b1ec22d

Browse files
bug fix #2008 (#2039)
1 parent cfa846e commit b1ec22d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

unsloth/models/llama.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1825,7 +1825,7 @@ def from_pretrained(
18251825

18261826
# Convert to HF format
18271827
_, quant_state_dict = get_vllm_state_dict(llm, config = model_config)
1828-
model = convert_vllm_to_huggingface(quant_state_dict, model_config, dtype)
1828+
model = convert_vllm_to_huggingface(quant_state_dict, model_config, dtype, bnb_config)
18291829
model.vllm_engine = llm
18301830
model.fast_generate = model.vllm_engine.generate
18311831
model.fast_generate_batches = functools.partial(generate_batches, model.vllm_engine)

0 commit comments

Comments
 (0)