github
/
ChatGLM-6B
mirror of https://github.com/THUDM/ChatGLM-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update trainer

pull/518/head
rainatam 2023-04-10 19:40:21 +08:00
parent 2073ac75d4
commit 173ccd8d27
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ptuning/trainer.py
View File

@ -2830,7 +2830,7 @@ class Trainer:
for k, v in self.model.named_parameters():
if v.requires_grad:
filtered_state_dict[k] = state_dict[k]
self.model.save_pretrained(output_dir, state_dict=state_dict)
self.model.save_pretrained(output_dir, state_dict=filtered_state_dict)
if self.tokenizer is not None:
self.tokenizer.save_pretrained(output_dir)