Skip to content

Commit

Permalink
only run setup_quantized_peft_meta_for_training for qlora+fsdp
Browse files Browse the repository at this point in the history
  • Loading branch information
winglian committed Mar 8, 2024
1 parent b32de13 commit af885de
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/axolotl/utils/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -1085,7 +1085,7 @@ def load_lora(model, cfg, inference=False, config_only=False):

if rank == 0:
model.print_trainable_parameters()
else:
elif cfg.fsdp and cfg.adapter == "qlora":
setup_quantized_peft_meta_for_training(model)

return model, lora_config

0 comments on commit af885de

Please sign in to comment.