From f523a0894c83fee00c8f10a8b3435ccf86f707bf Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 25 May 2023 09:23:56 -0400 Subject: [PATCH] stray s --- src/axolotl/utils/models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/axolotl/utils/models.py b/src/axolotl/utils/models.py index ea7661c05..5b243bec4 100644 --- a/src/axolotl/utils/models.py +++ b/src/axolotl/utils/models.py @@ -363,9 +363,9 @@ def load_lora(model, cfg): ) bits = None - if cfg.load_in_4bits: + if cfg.load_in_4bit: bits = 4 - elif cfg.load_in_8bits: + elif cfg.load_in_8bit: bits = 8 linear_names = find_all_linear_names(bits, model) logging.info(f"found linear modules: {repr(linear_names)}")