foo-barrr commited on
Commit
7feda8e
·
verified ·
1 Parent(s): 0b719f1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -1
README.md CHANGED
@@ -18,8 +18,20 @@ tags:
18
 
19
  ### Model Description
20
 
21
- <!-- Provide a longer summary of what this model is. -->
22
 
 
 
 
 
 
 
 
 
 
 
 
 
23
 
24
 
25
  - **Developed by:** [More Information Needed]
 
18
 
19
  ### Model Description
20
 
21
+ LoRA finetuned Llama-2-7B-Chat on the Alpaca dataset, mixed with safety data.
22
 
23
+ Dataset : https://github.com/vinid/safety-tuned-llamas/blob/main/data/training/saferpaca_Instructions_500.json
24
+
25
+ Base model : https://huggingface.co/TheBloke/Llama-2-7B-Chat-fp16
26
+
27
+ https://github.com/rahuln32/LLMs-Finetuning-Safety-Drift-Fixes/tree/main/llama2
28
+
29
+ Params :
30
+
31
+ Batch size : 64
32
+ Gradient accumulation steps : 1
33
+ LR : 5e-5
34
+ Epochs : 1
35
 
36
 
37
  - **Developed by:** [More Information Needed]