ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-01-04 06:32:00 +00:00
commit aa7fcda65c
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/models/loader.py
View file

@ -625,7 +625,9 @@ class FastLanguageModel(FastLlamaModel):
else:
# Preserve BitsAndBytesConfig if provided
if hasattr(quantization_config, "to_dict"):
model.config.update({"quantization_config": quantization_config.to_dict()})
model.config.update(
{"quantization_config": quantization_config.to_dict()}
)
elif isinstance(quantization_config, dict):
model.config.update({"quantization_config": quantization_config})