fix: config + ignore pkl

1 year ago · b66f127ade
parent 147c2fd7eb
commit b66f127ade
2 changed files with 3 additions and 2 deletions
--- a/.gitignore
+++ b/.gitignore
@ -1,3 +1,4 @@
+*.pkl
 ckpts*
 .deepspeed_env
 *.jsonl
--- a/configs/train/finetune_gptj_lora.yaml
+++ b/configs/train/finetune_gptj_lora.yaml
@ -1,6 +1,6 @@
 # model/tokenizer
-model_name: "EleutherAI/gpt-j-6B"
-tokenizer_name: "EleutherAI/gpt-j-6B"
+model_name: "EleutherAI/gpt-j-6b"
+tokenizer_name: "EleutherAI/gpt-j-6b"
 gradient_checkpointing: false
 save_name: "nomic-ai/gpt4all-mosaic"