AI/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2024-10-01 01:06:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update finetune.yaml

Browse Source
This commit is contained in:
Zach Nussbaum 2023-04-13 18:04:30 -07:00 committed by GitHub
parent 8325767c50
commit b744df7605
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configs/train/finetune.yaml
View File

@ -2,7 +2,7 @@
model_name: # add model here model_name: # add model here
tokenizer_name: # add model here tokenizer_name: # add model here
gradient_checkpointing: true gradient_checkpointing: true
save_name: "nomic-ai/gpt4all-full-multi-turn" save_name: # CHANGE
# dataset # dataset
streaming: false streaming: false
@ -16,7 +16,7 @@ lr: 5.0e-5
eval_every: 800 eval_every: 800
eval_steps: 100 eval_steps: 100
save_every: 800 save_every: 800
output_dir: "ckpts/gpt4all-full-multi" output_dir: # CHANGE
checkpoint: null checkpoint: null
lora: false lora: false
warmup_steps: 100 warmup_steps: 100