Views
No views yet
Service: Private
Node: StormPeak
GPU: 2 x RTX 4090 (24 GiB)
CPU: 62 vCPU
RAM: 400 GiB0.6.01base_model: ./outputs/out/DeepSeek-V2-Lite-Chat-Uncensored
2
3trust_remote_code: true
4
5load_in_8bit: false
6load_in_4bit: false
7strict: false
8
9chat_template: deepseek_v2
10rl: dpo
11datasets:
12 - path: /root/GreatFirewall-DPO/greatfirewall-dpo-v2_merged.json
13 data_files:
14 - /root/GreatFirewall-DPO/greatfirewall-dpo-v2_merged.json
15 ds_type: json
16 split: train
17 type:
18 field_prompt: prompt
19 field_chosen: chosen
20 field_rejected: rejected
21
22dataset_prepared_path:
23val_set_size: 0.05
24output_dir: ./outputs/out/DeepSeek-V2-Lite-Chat-Uncensored-Unbiased
25save_safetensors: true
26
27sequence_len: 4096
28sample_packing: false
29pad_to_sequence_len: true
30
31adapter: lora
32lora_model_dir:
33lora_r: 32
34lora_alpha: 16
35lora_dropout: 0.05
36lora_target_linear: true
37lora_fan_in_fan_out:
38
39gradient_accumulation_steps: 4
40micro_batch_size: 1
41num_epochs: 6
42optimizer: adamw_torch_fused
43lr_scheduler: cosine
44learning_rate: 0.0002
45
46train_on_inputs: false
47group_by_length: false
48bf16: true
49tf32: true
50
51gradient_checkpointing: true
52gradient_checkpointing_kwargs:
53 use_reentrant: true
54early_stopping_patience:
55resume_from_checkpoint:
56auto_resume_from_checkpoints: true
57logging_steps: 1
58flash_attention: true
59
60warmup_steps: 10
61evals_per_epoch: 4
62eval_table_size: 20
63eval_max_new_tokens: 128
64saves_per_epoch: 4
65save_total_limit: 20
66debug:
67deepspeed:
68weight_decay: 0.0
69fsdp:
70 - full_shard
71 - auto_wrap
72fsdp_config:
73 fsdp_limit_all_gathers: true
74 fsdp_sync_module_states: true
75 fsdp_offload_params: true
76 fsdp_use_orig_params: false
77 fsdp_cpu_ram_efficient_loading: true
78 fsdp_auto_wrap_policy: TRANSFORMER_BASED_WRAP
79 fsdp_transformer_layer_cls_to_wrap: DeepseekV2DecoderLayer
80 fsdp_state_dict_type: FULL_STATE_DICT
81 fsdp_sharding_strategy: FULL_SHARD
82special_tokens:
831@inproceedings{rafailov2023direct,
2 title = {{Direct Preference Optimization: Your Language Model is Secretly a Reward Model}},
3 author = {Rafael Rafailov and Archit Sharma and Eric Mitchell and Christopher D. Manning and Stefano Ermon and Chelsea Finn},
4 year = 2023,
5 booktitle = {Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, NeurIPS 2023, New Orleans, LA, USA, December 10 - 16, 2023},
6 url = {http://papers.nips.cc/paper_files/paper/2023/hash/a85b405ed65c6477a4fe8302b5e06ce7-Abstract-Conference.html},
7 editor = {Alice Oh and Tristan Naumann and Amir Globerson and Kate Saenko and Moritz Hardt and Sergey Levine},
8}1@misc{vonwerra2022trl,
2 title = {{TRL: Transformer Reinforcement Learning}},
3 author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallouédec},
4 year = 2020,
5 journal = {GitHub repository},
6 publisher = {GitHub},
7 howpublished = {\url{https://github.com/huggingface/trl}}
8}