Update README.md
Browse files
README.md
CHANGED
|
@@ -1,21 +1,54 @@
|
|
| 1 |
---
|
| 2 |
library_name: peft
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
| 4 |
-
## Training procedure
|
| 5 |
|
|
|
|
| 6 |
|
| 7 |
-
|
| 8 |
-
-
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
- llm_int8_threshold: 6.0
|
| 12 |
-
- llm_int8_skip_modules: None
|
| 13 |
-
- llm_int8_enable_fp32_cpu_offload: False
|
| 14 |
-
- llm_int8_has_fp16_weight: False
|
| 15 |
-
- bnb_4bit_quant_type: nf4
|
| 16 |
-
- bnb_4bit_use_double_quant: True
|
| 17 |
-
- bnb_4bit_compute_dtype: bfloat16
|
| 18 |
-
### Framework versions
|
| 19 |
|
|
|
|
| 20 |
|
| 21 |
-
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
library_name: peft
|
| 3 |
+
tags:
|
| 4 |
+
- tiiuae
|
| 5 |
+
- code
|
| 6 |
+
- instruct
|
| 7 |
+
- databricks-dolly-15k
|
| 8 |
+
- falcon-40b
|
| 9 |
+
datasets:
|
| 10 |
+
- databricks/databricks-dolly-15k
|
| 11 |
+
base_model: tiiuae/falcon-40b
|
| 12 |
---
|
|
|
|
| 13 |
|
| 14 |
+
For our finetuning process, we utilized the tiiuae/falcon-40b model and the Databricks-dolly-15k dataset.
|
| 15 |
|
| 16 |
+
This dataset, a meticulous compilation of over 15,000 records, was a result of the dedicated work of thousands of Databricks professionals. It was specifically designed to further improve the interactive capabilities of ChatGPT-like systems.
|
| 17 |
+
The dataset contributors crafted prompt / response pairs across eight distinct instruction categories. Besides the seven categories mentioned in the InstructGPT paper, they also ventured into an open-ended, free-form category. The contributors, emphasizing genuine and original content, refrained from sourcing information online, except in special cases where Wikipedia was the source for certain instruction categories. There was also a strict directive against the use of generative AI for crafting instructions or responses.
|
| 18 |
+
The contributors could address questions from their peers. Rephrasing the original question was encouraged, and there was a clear preference to answer only those queries they were certain about.
|
| 19 |
+
In some categories, the data comes with reference texts sourced from Wikipedia. Users might find bracketed Wikipedia citation numbers (like [42]) within the context field of the dataset. For smoother downstream applications, it's advisable to exclude these.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |
+
Our finetuning was conducted using the [MonsterAPI](https://monsterapi.ai)'s intuitive, no-code [LLM finetuner](https://docs.monsterapi.ai/fine-tune-a-large-language-model-llm).
|
| 22 |
|
| 23 |
+
Highlighting the cost-effectiveness and efficiency of the process,
|
| 24 |
+
the entire session was finished in just 5 hours and 40 minutes, leveraging an A6000 48GB GPU.
|
| 25 |
+
The total cost for this efficient run was a mere `$11.8`.
|
| 26 |
+
|
| 27 |
+
#### Hyperparameters & Run details:
|
| 28 |
+
- Epochs: 1
|
| 29 |
+
- Cost: $11.8
|
| 30 |
+
- Model Path: tiiuae/falcon-40b
|
| 31 |
+
- Dataset: databricks/databricks-dolly-15k
|
| 32 |
+
- Learning rate: 0.0002
|
| 33 |
+
- Data split: Training 90% / Validation 10%
|
| 34 |
+
- Gradient accumulation steps: 4
|
| 35 |
+
|
| 36 |
+
license: apache-2.0
|
| 37 |
+
---
|
| 38 |
+
|
| 39 |
+
######
|
| 40 |
+
|
| 41 |
+
Prompt Used:
|
| 42 |
+
|
| 43 |
+
### INSTRUCTION:
|
| 44 |
+
[instruction]
|
| 45 |
+
|
| 46 |
+
[context]
|
| 47 |
+
|
| 48 |
+
### RESPONSE:
|
| 49 |
+
[response]
|
| 50 |
+
|
| 51 |
+
Loss metrics
|
| 52 |
+
|
| 53 |
+
Training loss (Blue) Validation Loss (orange):
|
| 54 |
+

|