From 99095573c3a510f6170b68d697d55bd13f5021df Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sat, 3 May 2025 02:46:50 -0400 Subject: [PATCH] add tabs back to code check --- src/axolotl/monkeypatch/peft/utils.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/src/axolotl/monkeypatch/peft/utils.py b/src/axolotl/monkeypatch/peft/utils.py index fed88a0ed..d777128b1 100644 --- a/src/axolotl/monkeypatch/peft/utils.py +++ b/src/axolotl/monkeypatch/peft/utils.py @@ -12,19 +12,19 @@ from axolotl.monkeypatch.utils import detab_code LOG = logging.getLogger(__name__) ORIGINAL_PREPARE_CODE = """ - for param in model.parameters(): - if ( - (param.dtype == torch.float16) or (param.dtype == torch.bfloat16) - ) and param.__class__.__name__ != "Params4bit": - param.data = param.data.to(torch.float32) + for param in model.parameters(): + if ( + (param.dtype == torch.float16) or (param.dtype == torch.bfloat16) + ) and param.__class__.__name__ != "Params4bit": + param.data = param.data.to(torch.float32) """ PATCHED_PREPARE_CODE = """ - for name, param in model.named_parameters(): - if ( - (param.dtype == torch.float16) or (param.dtype == torch.bfloat16) - ) and param.__class__.__name__ != "Params4bit" and "norm" in name: - param.data = param.data.to(torch.float32) + for name, param in model.named_parameters(): + if ( + (param.dtype == torch.float16) or (param.dtype == torch.bfloat16) + ) and param.__class__.__name__ != "Params4bit" and "norm" in name: + param.data = param.data.to(torch.float32) """