Vasanth commited on
Commit
b19a487
·
verified ·
1 Parent(s): 4cf84ee

Model save

Browse files
Files changed (1) hide show
  1. README.md +7 -5
README.md CHANGED
@@ -5,18 +5,20 @@ tags:
5
  - trl
6
  - sft
7
  - generated_from_trainer
 
 
8
  base_model: 01-ai/Yi-9B
9
  model-index:
10
- - name: ti-text-to-sql
11
  results: []
12
  ---
13
 
14
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
  should probably proofread and complete it, then remove this comment. -->
16
 
17
- # ti-text-to-sql
18
 
19
- This model is a fine-tuned version of [01-ai/Yi-9B](https://huggingface.co/01-ai/Yi-9B) on the None dataset.
20
 
21
  ## Model description
22
 
@@ -41,7 +43,7 @@ The following hyperparameters were used during training:
41
  - seed: 42
42
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
  - lr_scheduler_type: cosine
44
- - training_steps: 250
45
  - mixed_precision_training: Native AMP
46
 
47
  ### Training results
@@ -51,7 +53,7 @@ The following hyperparameters were used during training:
51
  ### Framework versions
52
 
53
  - PEFT 0.9.0
54
- - Transformers 4.37.2
55
  - Pytorch 2.1.0+cu121
56
  - Datasets 2.18.0
57
  - Tokenizers 0.15.2
 
5
  - trl
6
  - sft
7
  - generated_from_trainer
8
+ datasets:
9
+ - generator
10
  base_model: 01-ai/Yi-9B
11
  model-index:
12
+ - name: yi-text-to-sql
13
  results: []
14
  ---
15
 
16
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
  should probably proofread and complete it, then remove this comment. -->
18
 
19
+ # yi-text-to-sql
20
 
21
+ This model is a fine-tuned version of [01-ai/Yi-9B](https://huggingface.co/01-ai/Yi-9B) on the generator dataset.
22
 
23
  ## Model description
24
 
 
43
  - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: cosine
46
+ - num_epochs: 1
47
  - mixed_precision_training: Native AMP
48
 
49
  ### Training results
 
53
  ### Framework versions
54
 
55
  - PEFT 0.9.0
56
+ - Transformers 4.38.2
57
  - Pytorch 2.1.0+cu121
58
  - Datasets 2.18.0
59
  - Tokenizers 0.15.2