diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index e4d33e70f6..a5b7d18164 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -858,7 +858,7 @@ class FastModel(FastBaseModel): trust_remote_code = trust_remote_code, ) # Patch it as well! - model = FastBaseModel.post_patch_model(model, use_gradient_checkpointing,trust_remote_code = trust_remote_code) + model = FastBaseModel.post_patch_model(model, use_gradient_checkpointing, trust_remote_code = trust_remote_code) pass return model, tokenizer pass