tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

stray s

Browse Source
This commit is contained in:
Wing Lian
2023-05-25 09:23:56 -04:00
parent 676d7da661
commit f523a0894c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/axolotl/utils/models.py
View File

@@ -363,9 +363,9 @@ def load_lora(model, cfg):
) )
bits = None bits = None
if cfg.load_in_4bits: if cfg.load_in_4bit:
bits = 4 bits = 4
elif cfg.load_in_8bits: elif cfg.load_in_8bit:
bits = 8 bits = 8
linear_names = find_all_linear_names(bits, model) linear_names = find_all_linear_names(bits, model)
logging.info(f"found linear modules: {repr(linear_names)}") logging.info(f"found linear modules: {repr(linear_names)}")