chore: lint

axolotl-ai-cloud · Mar 16, 2024 · c9fe327 · c9fe327
1 parent a914cb3
commit c9fe327
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/axolotl/utils/models.py b/src/axolotl/utils/models.py
@@ -888,7 +888,9 @@ def load_and_quantize_parallel(name_param, model, **kwargs):
 
     if cfg.adapter in ["lora", "qlora"]:
         if cfg.gradient_checkpointing:
-            model.gradient_checkpointing_enable(gradient_checkpointing_kwargs=cfg.gradient_checkpointing_kwargs)
+            model.gradient_checkpointing_enable(
+                gradient_checkpointing_kwargs=cfg.gradient_checkpointing_kwargs
+            )
         if (
             cfg.load_in_8bit or cfg.load_in_4bit
         ) and not skip_prepare_model_for_kbit_training: