CyberNative commited on
Commit
d3f16ed
1 Parent(s): 950eeb4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -13,6 +13,7 @@ Base cybersecurity model for future fine-tuning, it is not recomended to use on
13
  - **lora_r:** 128
14
  - **lora_alpha:** 16
15
  - **num_epochs:** 3
 
16
 
17
  [<img src="https://raw.githubusercontent.com/OpenAccess-AI-Collective/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/OpenAccess-AI-Collective/axolotl)
18
 
 
13
  - **lora_r:** 128
14
  - **lora_alpha:** 16
15
  - **num_epochs:** 3
16
+ - **flash_attention:** true (FlashAttention-2)
17
 
18
  [<img src="https://raw.githubusercontent.com/OpenAccess-AI-Collective/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/OpenAccess-AI-Collective/axolotl)
19