gpt4all/gpt4all-training/configs/inference/gptj.yaml

15 lines
243 B
YAML
Raw Normal View History

2023-04-07 01:40:39 +00:00
# model/tokenizer
2023-04-10 02:14:20 +00:00
model_name: "nomic-ai/gpt4all-warmup-lr-epoch_1"
2023-04-07 01:40:39 +00:00
tokenizer_name: "EleutherAI/gpt-j-6B"
# dataset
streaming: false
num_proc: 64
2023-04-10 02:14:20 +00:00
dataset_path: "nomic-ai/turbo-500k-multi"
2023-04-07 01:40:39 +00:00
max_length: 1024
batch_size: 32
# logging
seed: 42