File size: 484 Bytes
8366b03
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
models:
    openllama:
        model_name: OpenLLAMAModel
        agent_name: DeepSpeedAgent
        stage1_train_dataset: SupervisedDataset
        test_dataset: SelfInstructTestDataset
    openllama_peft:
        model_name: OpenLLAMAPEFTModel
        agent_name: DeepSpeedAgent
        stage1_train_dataset: SupervisedDataset
        test_dataset: SelfInstructTestDataset

# ========= Global configuration ========== #
logging_step: 5
# ========= Global configuration ========== #