ilotoki_thu
/
alpaca_lora_4bit
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix bug

This commit is contained in:
John Smith 2023-04-17 13:42:50 +08:00
parent 7a71b0dd12
commit e64ff9facd
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
finetune.py
View File

@ -172,6 +172,9 @@ if not ft_config.skip:
else: else:
trainer.train() trainer.train()
# Restore old model state dict
model.state_dict = old_state_dict
print('Train completed.') print('Train completed.')
# Save Model # Save Model