Views
No views yet

0.3.01base_model: deepseek-ai/deepseek-coder-1.3b-base
2model_type: AutoModelForCausalLM
3trust_remote_code: true
4load_in_8bit: false
5load_in_4bit: false
6strict: false
7
8
9datasets:
10 - path: CodeGPTPlus/typescript-0-500000-seq1024
11 type: completion
12 field: text
13
14
15val_set_size: 0.001
16output_dir: ./fft-out
17
18sequence_len: 1024
19
20adapter:
21lora_model_dir:
22lora_r:
23lora_alpha:
24lora_dropout:
25lora_target_linear:
26lora_fan_in_fan_out:
27lora_modules_to_save:
28
29wandb_project: deepseek_1.3_fft
30wandb_entity:
31wandb_watch:
32wandb_name: aws_a10g
33wandb_log_model: end
34
35
36gradient_accumulation_steps: 2
37micro_batch_size: 20
38num_epochs: 1
39optimizer: adamw_bnb_8bit
40adam_beta1: 0.9
41adam_beta2: 0.999
42adam_epsilon: 0.000001
43max_grad_norm: 1.0
44weight_decay: 0.1
45lr_scheduler: cosine
46learning_rate: 0.00002
47train_on_inputs: false
48group_by_length: false
49bf16: true
50fp16: false
51tf32: false
52gradient_checkpointing: true
53early_stopping_patience:
54resume_from_checkpoint:
55local_rank:
56logging_steps: 1
57xformers_attention:
58flash_attention: true
59
60loss_watchdog_threshold: 5.0
61loss_watchdog_patience: 3
62
63hub_model_id: CodeGPTPlus/deepseek_coder_1.3b_typescript
64hub_strategy: every_save
65warmup_ratio: 0.01
66evals_per_epoch: 20
67saves_per_epoch: 3
68debug:
69deepspeed:
70
71fsdp:
72fsdp_config:
73special_tokens:
74 bos_token: "<|begin▁of▁sentence|>"
75 eos_token: "<|end▁of▁sentence|>"
76 pad_token: "<|end▁of▁sentence|>"1from transformers import AutoTokenizer, AutoModelForCausalLM
2tokenizer = AutoTokenizer.from_pretrained("CodeGPTPlus/deepseek-coder-1.3b-typescript",
3 trust_remote_code=True)
4model = AutoModelForCausalLM.from_pretrained("CodeGPTPlus/deepseek-coder-1.3b-typescript",
5 trust_remote_code=True).cuda()
6
7input_text = """<|fim▁begin|>function quickSort(arr: number[]): number[] {
8 if (arr.length <= 1) {
9 return arr;
10 }
11 const pivot = arr[0];
12 const left = [];
13 const right = [];
14<|fim▁hole|>
15 return [...quickSort(left), pivot, ...quickSort(right)];
16}<|fim▁end|>"""
17
18inputs = tokenizer(input_text, return_tensors="pt").to(model.device)
19outputs = model.generate(**inputs, max_length=256)
20print(tokenizer.decode(outputs[0], skip_special_tokens=True))ollama run codegpt/deepseek-coder-1.3b-typescript
1<|fim▁begin|>function quickSort(arr: number[]): number[] {
2 if (arr.length <= 1) {
3 return arr;
4 }
5 const pivot = arr[0];
6 const left = [];
7 const right = [];
8<|fim▁hole|>
9 return [...quickSort(left), pivot, ...quickSort(right)];
10}<|fim▁end|>| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 1.0745 | 0.0 | 1 | 0.8681 |
| 1.2267 | 0.05 | 1308 | 0.8130 |
| 1.1594 | 0.1 | 2616 | 0.8018 |
| 0.7674 | 0.15 | 3924 | 0.7942 |
| 0.6443 | 0.2 | 5232 | 0.7889 |
| 0.9155 | 0.25 | 6540 | 0.7847 |
| 0.7501 | 0.3 | 7848 | 0.7819 |
| 0.8835 | 0.35 | 9156 | 0.7792 |
| 0.7261 | 0.4 | 10464 | 0.7769 |
| 0.9746 | 0.45 | 11772 | 0.7748 |
| 0.6884 | 0.5 | 13080 | 0.7734 |
| 0.6104 | 0.55 | 14388 | 0.7722 |
| 0.8876 | 0.6 | 15696 | 0.7710 |
| 0.9567 | 0.65 | 17004 | 0.7703 |
| 0.6915 | 0.7 | 18312 | 0.7696 |
| 0.8874 | 0.75 | 19620 | 0.7691 |
| 0.6124 | 0.8 | 20928 | 0.7686 |
| 0.8147 | 0.85 | 22236 | 0.7684 |
| 0.8021 | 0.9 | 23544 | 0.7683 |
| 0.8665 | 0.95 | 24852 | 0.7681 |