diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py
index 9b58282f2a..152a7cd69d 100644
--- a/unsloth/models/loader.py
+++ b/unsloth/models/loader.py
@@ -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})