Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

ft.yml 1.4 KB

You have to be logged in to leave a comment. Sign In
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
  1. base_model: tiiuae/falcon-7b
  2. base_model_config: tiiuae/falcon-7b
  3. trust_remote_code: true
  4. model_type: AutoModelForCausalLM
  5. tokenizer_type: AutoTokenizer
  6. load_in_8bit: false
  7. load_in_4bit: false
  8. gptq: false
  9. strict: false
  10. push_dataset_to_hub:
  11. datasets:
  12. - path: teknium/GPT4-LLM-Cleaned
  13. type: alpaca:chat
  14. dataset_prepared_path: last_run_prepared
  15. val_set_size: 0.01
  16. adapter:
  17. lora_model_dir:
  18. sequence_len: 2048
  19. max_packed_sequence_len:
  20. lora_r:
  21. lora_alpha:
  22. lora_dropout:
  23. lora_target_modules:
  24. lora_target_linear:
  25. lora_fan_in_fan_out:
  26. wandb_project:
  27. wandb_watch:
  28. wandb_run_id:
  29. wandb_log_model:
  30. output_dir: ./ft-out
  31. micro_batch_size: 1
  32. gradient_accumulation_steps: 2
  33. num_epochs: 4
  34. optimizer: adamw_bnb_8bit
  35. torchdistx_path:
  36. lr_scheduler: cosine
  37. learning_rate: 0.0002
  38. train_on_inputs: false
  39. group_by_length: false
  40. bf16: true
  41. fp16: false
  42. tf32: true
  43. gradient_checkpointing: true
  44. # stop training after this many evaluation losses have increased in a row
  45. # https://huggingface.co/transformers/v4.2.2/_modules/transformers/trainer_callback.html#EarlyStoppingCallback
  46. early_stopping_patience: 3
  47. resume_from_checkpoint:
  48. auto_resume_from_checkpoints: true
  49. local_rank:
  50. logging_steps: 1
  51. xformers_attention: true
  52. flash_attention:
  53. gptq_groupsize:
  54. gptq_model_v1:
  55. warmup_steps: 10
  56. eval_steps: 5
  57. save_steps: 10
  58. debug:
  59. deepspeed:
  60. weight_decay: 0.000001
  61. fsdp:
  62. fsdp_config:
  63. special_tokens:
  64. pad_token: "<|endoftext|>"
  65. bos_token: ">>ABSTRACT<<"
  66. eos_token: "<|endoftext|>"
Tip!

Press p or to see the previous file or, n or to see the next file

Comments

Loading...