浏览代码

fix the save_train_param condition

Hamid Shojanazeri 1 年之前
父节点
当前提交
88d3e1febc
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      utils/train_utils.py

+ 1 - 1
utils/train_utils.py

@@ -206,7 +206,7 @@ def train(model, train_dataloader,eval_dataloader, tokenizer, optimizer, lr_sche
         results['avg_eval_loss'] = avg_eval_loss
         
     #saving the training params including fsdp setting for reference.
-    if train_config.enable_fsdp and fsdp_config:
+    if train_config.enable_fsdp and not train_config.use_peft:
         save_train_params(train_config, fsdp_config, rank)
         
     return results