Views
No views yet
0.9.21base_model: byroneverson/gemma-2-27b-it-abliterated
2load_in_8bit: false
3load_in_4bit: true
4adapter: qlora
5wandb_name: gemma-2-27b-it-abliterated-chem-claude-1-comp3-sort-pate-self
6output_dir: ./outputs/out/gemma-2-27b-it-abliterated-chem-claude-1-comp3-sort-pate-self
7hub_model_id: cgifbribcgfbi/gemma-2-27b-it-abliterated-chem-claude-1-comp3-sort-pate-self
8
9tokenizer_type: AutoTokenizer
10push_dataset_to_hub:
11strict: false
12
13datasets:
14 - path: dset_comp3.0_sortpatent_count_pat400_in1_num5000_gemma2_5000.jsonl
15 type: chat_template
16 field_messages: messages
17
18dataset_prepared_path: last_run_prepared
19# val_set_size: 0.05
20# eval_sample_packing: False
21save_safetensors: true
22
23sequence_len: 3015
24sample_packing: true
25pad_to_sequence_len: true
26
27lora_r: 64
28lora_alpha: 32
29lora_dropout: 0.05
30lora_target_modules:
31lora_target_linear: true
32
33wandb_mode:
34wandb_project: finetune-sweep
35wandb_entity: gpoisjgqetpadsfke
36wandb_watch:
37wandb_run_id:
38wandb_log_model:
39
40gradient_accumulation_steps: 1
41micro_batch_size: 4 # This will be automatically adjusted based on available GPU memory
42num_epochs: 4
43optimizer: adamw_torch_fused
44lr_scheduler: cosine
45learning_rate: 0.00002
46
47train_on_inputs: false
48group_by_length: true
49bf16: true
50tf32: true
51
52gradient_checkpointing: true
53gradient_checkpointing_kwargs:
54 use_reentrant: true
55logging_steps: 1
56flash_attention: true
57
58warmup_steps: 10
59evals_per_epoch: 3
60saves_per_epoch: 1
61weight_decay: 0.01
62fsdp:
63 - full_shard
64 - auto_wrap
65fsdp_config:
66 fsdp_limit_all_gathers: true
67 fsdp_sync_module_states: true
68 fsdp_offload_params: false
69 fsdp_use_orig_params: false
70 fsdp_cpu_ram_efficient_loading: true
71 fsdp_auto_wrap_policy: TRANSFORMER_BASED_WRAP
72 fsdp_transformer_layer_cls_to_wrap: Gemma2DecoderLayer
73 fsdp_state_dict_type: FULL_STATE_DICT
74 fsdp_sharding_strategy: FULL_SHARD
75special_tokens:
76 pad_token: <|finetune_right_pad_id|>
77