LH-Tech-AI commited on
Commit
7db491e
·
verified ·
1 Parent(s): 14ad7d8

Update finetune-apex-1.6.py

Browse files
Files changed (1) hide show
  1. finetune-apex-1.6.py +2 -2
finetune-apex-1.6.py CHANGED
@@ -6,8 +6,8 @@ from model import GPTConfig, GPT
6
 
7
  import numpy as np
8
 
9
- out_dir = '/media/leo/Data/checkpoints/Apex_1.6'
10
- init_from = '/media/leo/Data/checkpoints/350m_fineweb'
11
  dataset = 'apex-1.6-sft'
12
 
13
  batch_size = 4
 
6
 
7
  import numpy as np
8
 
9
+ out_dir = 'your_out_dir'
10
+ init_from = 'your_init_from_dir'
11
  dataset = 'apex-1.6-sft'
12
 
13
  batch_size = 4