Views
No views yet
0.4.11base_model: UnfilteredAI/NSFW-3B
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4
5load_in_8bit: false
6load_in_4bit: true
7strict: false
8
9datasets:
10 - path: Fischerboot/freedom-rp-alpaca-shortend
11 type: alpaca
12 - path: Fischerboot/mongotom-40k-alpaca
13 type: alpaca
14 - path: TokenBender/roleplay_alpaca
15 type: alpaca
16 ds_type: parquet
17dataset_prepared_path:
18val_set_size: 0.05
19output_dir: ./outputs/32r
20
21adapter: qlora
22lora_model_dir:
23
24sequence_len: 4096
25sample_packing: true
26eval_sample_packing: false
27pad_to_sequence_len: true
28
29lora_r: 32
30lora_alpha: 16
31lora_dropout: 0.05
32lora_target_modules:
33lora_target_linear: true
34lora_fan_in_fan_out:
35
36wandb_project:
37wandb_entity:
38wandb_watch:
39wandb_name:
40wandb_log_model:
41
42gradient_accumulation_steps: 4
43micro_batch_size: 2
44num_epochs: 2
45optimizer: paged_adamw_32bit
46lr_scheduler: cosine
47learning_rate: 0.0002
48
49train_on_inputs: false
50group_by_length: false
51bf16: auto
52fp16:
53tf32: false
54
55gradient_checkpointing: true
56early_stopping_patience:
57resume_from_checkpoint:
58local_rank:
59logging_steps: 1
60xformers_attention:
61flash_attention: true
62
63warmup_steps: 10
64evals_per_epoch: 1
65saves_per_epoch: 1
66debug:
67deepspeed:
68weight_decay: 0.0
69fsdp:
70fsdp_config:
71special_tokens:
72trust_remote_code: true| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 8.0268 | 0.0032 | 1 | 3.2872 |
| 2.2591 | 1.0 | 314 | 2.1557 |
| 2.045 | 1.9753 | 628 | 2.1202 |