Merge pull request #275 from NanoCode012/feat/safetensors

Feat: Add save_safetensors
This commit is contained in:
NanoCode012
2023-07-14 23:07:26 +09:00
committed by GitHub
2 changed files with 6 additions and 0 deletions

View File

@@ -413,6 +413,9 @@ logging_steps:
save_steps:
eval_steps:
# save model as safetensors (require safetensors package)
save_safetensors:
# whether to mask out or include the human's prompt from the training labels
train_on_inputs: false
# don't use this, leads to wonky training (according to someone on the internet)